基于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 import用法以及与from...import的区别
May 28 Python
Python对List中的元素排序的方法
Apr 01 Python
Python3.5基础之变量、数据结构、条件和循环语句、break与continue语句实例详解
Apr 26 Python
PyTorch的深度学习入门之PyTorch安装和配置
Jun 27 Python
Python实现最常见加密方式详解
Jul 13 Python
关于Python形参打包与解包小技巧分享
Aug 24 Python
基于python+selenium的二次封装的实现
Jan 06 Python
Python栈的实现方法示例【列表、单链表】
Feb 22 Python
Python tkinter 下拉日历控件代码
Mar 04 Python
使用python批量转换文件编码为UTF-8的实现
Apr 03 Python
浅谈python多线程和多线程变量共享问题介绍
Apr 17 Python
python多进程使用函数封装实例
May 02 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
mysql建立外键
2006/11/25 PHP
PHP数组对比函数,存在交集则返回真,否则返回假
2011/02/03 PHP
php流量统计功能的实现代码
2012/09/29 PHP
基于php split()函数的用法详解
2013/06/05 PHP
Javascript 中文字符串处理额外注意事项
2009/11/15 Javascript
jquery中邮箱地址 URL网站地址正则验证实例代码
2013/09/15 Javascript
引用外部脚本时script标签关闭的写法
2014/01/20 Javascript
jQuery中:focus选择器用法实例
2014/12/30 Javascript
jquery仿百度百科底部浮动导航特效
2015/08/08 Javascript
JS+CSS相对定位实现的下拉菜单
2015/10/06 Javascript
使用DeviceOne实现微信小程序功能
2016/12/29 Javascript
javascript用rem来做响应式开发
2018/01/13 Javascript
vue.js select下拉框绑定和取值方法
2018/03/03 Javascript
Node.js log4js日志管理详解
2018/07/31 Javascript
vue+elementUI实现表格关键字筛选高亮
2020/10/26 Javascript
vue实现在线翻译功能
2019/09/27 Javascript
vue+AI智能机器人回复功能实现
2020/07/16 Javascript
Python语法快速入门指南
2015/10/12 Python
使用Python编写简单的画图板程序的示例教程
2015/12/08 Python
python中int与str互转方法
2018/07/02 Python
Python简单获取二维数组行列数的方法示例
2018/12/21 Python
Python Sympy计算梯度、散度和旋度的实例
2019/12/06 Python
python通用读取vcf文件的类(复制粘贴即可用)
2020/02/29 Python
使用pyecharts1.7进行简单的可视化大全
2020/05/17 Python
使用Keras 实现查看model weights .h5 文件的内容
2020/06/09 Python
HTML5 File接口在web页面上使用文件下载
2017/02/27 HTML / CSS
Melijoe英国官网:法国儿童时尚网站
2016/11/18 全球购物
美国旅游网站:Tours4Fun
2017/02/17 全球购物
澳大利亚在线消费电子产品商店:TobyDeals
2020/01/05 全球购物
文明餐桌行动实施方案
2014/02/19 职场文书
土建专业大学生自荐信范文
2014/04/09 职场文书
九一八事变演讲稿
2014/09/05 职场文书
四川省传达学习贯彻党的群众路线教育实践活动总结大会精神新闻稿
2014/10/26 职场文书
酒会邀请函
2015/01/31 职场文书
物流仓管员岗位职责
2015/04/01 职场文书
使用springMVC所需要的pom配置
2021/09/15 Java/Android