基于tensorflow指定GPU运行及GPU资源分配的几种方式小结


Posted in Python onFebruary 03, 2020

1. 在终端执行时设置使用哪些GPU(两种方式)

(1) 如下(export 语句执行一次就行了,以后再运行代码不用执行)

基于tensorflow指定GPU运行及GPU资源分配的几种方式小结

(2) 如下

基于tensorflow指定GPU运行及GPU资源分配的几种方式小结

2. 代码中指定(两种方式)

(1)

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "1"

(2)

# Creates a graph.
with tf.device('/gpu:1'):
 a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')
 b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b')
 c = tf.matmul(a, b)
# Creates a session with log_device_placement set to True.
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
# Runs the op.
print sess.run(c)

若想使用多个GPU,如下

c = []
for d in ['/gpu:0', '/gpu:1']:
 with tf.device(d):
  a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3])
  b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2])
  c.append(tf.matmul(a, b))
with tf.device('/cpu:0'):
 sum = tf.add_n(c)
# Creates a session with log_device_placement set to True.
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
# Runs the op.
print sess.run(sum)

3.GPU资源分配

(1) 设置允许GPU增长

config = tf.ConfigProto()
config.gpu_options.allow_growth = True
session = tf.Session(config=config, ...)

(2) 设置每个GPU内存使用多少

config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.4
session = tf.Session(config=config, ...)

以上这篇基于tensorflow指定GPU运行及GPU资源分配的几种方式小结就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python中使用Beautiful Soup库的超详细教程
Apr 30 Python
python基于BeautifulSoup实现抓取网页指定内容的方法
Jul 09 Python
Python 爬虫学习笔记之单线程爬虫
Sep 21 Python
Python3.6安装及引入Requests库的实现方法
Jan 24 Python
对python字典元素的添加与修改方法详解
Jul 06 Python
Python发送邮件功能示例【使用QQ邮箱】
Dec 04 Python
Django框架之DRF 基于mixins来封装的视图详解
Jul 23 Python
python单向循环链表原理与实现方法示例
Dec 03 Python
python3 常见解密加密算法实例分析【base64、MD5等】
Dec 19 Python
python3中使用__slots__限定实例属性操作分析
Feb 14 Python
Python爬取豆瓣数据实现过程解析
Oct 27 Python
用Python提取PDF表格的方法
Apr 11 Python
tensorflow指定GPU与动态分配GPU memory设置
Feb 03 #Python
TensorFlow内存管理bfc算法实例
Feb 03 #Python
python numpy数组中的复制知识解析
Feb 03 #Python
opencv python Canny边缘提取实现过程解析
Feb 03 #Python
Pycharm debug调试时带参数过程解析
Feb 03 #Python
Python使用enumerate获取迭代元素下标
Feb 03 #Python
TensorFlow 显存使用机制详解
Feb 03 #Python
You might like
PHP的foreach中使用引用时需要注意的一个问题和解决方法
2014/05/29 PHP
ThinkPHP视图查询详解
2014/06/30 PHP
又十个超级有用的PHP代码片段
2015/09/24 PHP
php接口实现拖拽排序功能
2018/04/23 PHP
PHP+Redis链表解决高并发下商品超卖问题(实现原理及步骤)
2020/08/03 PHP
JavaScript 读取元素的CSS信息的代码
2010/02/07 Javascript
js从10种颜色中随机取色实现每次取出不同的颜色
2013/10/23 Javascript
Javascript操作URL函数修改版
2013/11/07 Javascript
jquery easyui 结合jsp简单展现table数据示例
2014/04/18 Javascript
论Bootstrap3和Foundation5网格系统的异同
2016/05/16 Javascript
浅谈Web页面向后台提交数据的方式和选择
2016/09/23 Javascript
domReady的实现案例
2016/11/23 Javascript
深入浅析Vue组件开发
2016/11/25 Javascript
JS中事件冒泡和事件捕获介绍
2016/12/13 Javascript
bootstrap fileinput实现文件上传功能
2017/08/23 Javascript
解决Vue编译时写在style中的路径问题
2017/09/21 Javascript
vue实现在表格里,取每行的id的方法
2018/03/09 Javascript
vue添加axios,并且指定baseurl的方法
2018/09/19 Javascript
vue环形进度条组件实例应用
2018/10/10 Javascript
微信小程序后端(java)开发流程的详细步骤
2019/11/13 Javascript
小程序实现背景音乐播放和暂停
2020/06/19 Javascript
[13:55]Newbee vs Team Spirit
2018/06/07 DOTA
使用Python构建Hopfield网络的教程
2015/04/14 Python
python文件操作之目录遍历实例分析
2015/05/20 Python
python中不能连接超时的问题及解决方法
2018/06/10 Python
基于scrapy的redis安装和配置方法
2018/06/13 Python
基于python traceback实现异常的获取与处理
2019/12/13 Python
Python中如何将一个类方法变为多个方法
2019/12/30 Python
Timberland美国官网:全球领先的户外品牌
2016/08/15 全球购物
欧洲领先的技术商店:eibmarkt.com
2019/05/10 全球购物
银行毕业实习自我鉴定
2013/09/19 职场文书
保险专业大学生职业规划书
2014/03/03 职场文书
2015年班长个人工作总结
2015/04/03 职场文书
老公写给老婆的检讨书
2015/05/06 职场文书
2019年员工晋升管理制度范本!
2019/07/08 职场文书
Jupyter Notebook 如何修改字体和大小以及更改字体样式
2021/06/03 Python