基于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函数中定义参数的四种方式
Nov 30 Python
python logging日志模块以及多进程日志详解
Apr 18 Python
Python之pandas读写文件乱码的解决方法
Apr 20 Python
Python切片操作深入详解
Jul 27 Python
对python 合并 累加两个dict的实例详解
Jan 21 Python
Python参数解析模块sys、getopt、argparse使用与对比分析
Apr 02 Python
为什么是 Python -m
Jun 19 Python
python计算auc的方法
Sep 09 Python
Python字典实现伪切片功能
Oct 28 Python
详解python爬取弹幕与数据分析
Nov 14 Python
用python基于appium模块开发一个自动收取能量的小助手
Sep 25 Python
Python日志模块logging用法
Jun 05 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 error_log 函数的使用
2009/04/13 PHP
php实现的读取CSV文件函数示例
2017/02/07 PHP
PHP静态成员变量和非静态成员变量详解
2017/02/14 PHP
Laravel框架路由和控制器的绑定操作方法
2018/06/12 PHP
解决jquery中美元符号命名冲突问题
2014/01/08 Javascript
javascript之with的使用(阿里云、淘宝使用代码分析)
2016/10/11 Javascript
jQuery学习笔记之入门
2016/12/14 Javascript
canvas实现图像放大镜
2017/02/06 Javascript
vue使用element-ui的el-input监听不了回车事件的解决方法
2018/01/12 Javascript
js中el表达式的使用和非空判断方法
2018/03/28 Javascript
微信小程序自定义底部导航带跳转功能
2018/11/27 Javascript
jQuery简单实现根据日期计算星期几的方法
2019/01/09 jQuery
React如何实现浏览器打印部分内容详析
2019/05/19 Javascript
[47:45]Liquid vs OG 2018国际邀请赛小组赛BO2 第二场 8.16
2018/08/17 DOTA
Python fileinput模块使用介绍
2014/11/30 Python
pymongo为mongodb数据库添加索引的方法
2015/05/11 Python
Python读取properties配置文件操作示例
2018/03/29 Python
Python读写docx文件的方法
2018/05/08 Python
对Python中创建进程的两种方式以及进程池详解
2019/01/14 Python
Python I/O与进程的详细讲解
2019/03/08 Python
Python实现的对一个数进行因式分解操作示例
2019/06/27 Python
Python使用itchat 功能分析微信好友性别和位置
2019/08/05 Python
Django 实现Admin自动填充当前用户的示例代码
2019/11/18 Python
基于python使用tibco ems代码实例
2019/12/20 Python
Python PyPDF2模块安装使用解析
2020/01/19 Python
意大利在线药房:shop-farmacia.it
2019/03/12 全球购物
string = null 和string = ''的区别
2013/04/28 面试题
Linux如何命名文件--使用文件名时应注意
2014/05/29 面试题
历史学专业毕业生求职信
2013/09/27 职场文书
服务员自我评价
2014/01/25 职场文书
中华魂放飞梦想演讲稿
2014/08/26 职场文书
红高粱观后感
2015/06/10 职场文书
2016春季幼儿园大班开学寄语
2015/12/03 职场文书
《为人民服务》教学反思
2016/02/20 职场文书
2016年“6.26”禁毒宣传月系列活动总结
2016/04/05 职场文书
升级 Win11 还是坚守 Win10?微软 Win11 新系统缺失功能大盘点
2022/04/05 数码科技