基于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中面向对象编程的相关知识
May 25 Python
Python中的descriptor描述器简明使用指南
Jun 02 Python
利用Python在一个文件的头部插入数据的实例
May 02 Python
Python3实现转换Image图片格式
Jun 21 Python
ubuntu16.04制作vim和python3的开发环境
Sep 23 Python
Python的iOS自动化打包实例代码
Nov 22 Python
解决Django Static内容不能加载显示的问题
Jul 28 Python
Python Gitlab Api 使用方法
Aug 28 Python
调试Django时打印SQL语句的日志代码实例
Sep 12 Python
详解Python中打乱列表顺序random.shuffle()的使用方法
Nov 11 Python
使用python处理题库表格并转化为word形式的实现
Apr 14 Python
python 爬取天气网卫星图片
Jun 07 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_MySQL教程-第一天
2007/03/18 PHP
php学习笔记之 函数声明(二)
2011/06/09 PHP
php日历制作代码分享
2014/01/20 PHP
在PHP中输出JS语句以及乱码问题的解决方案
2019/02/13 PHP
Node.js与Sails ~项目结构与Mvc实现及日志机制
2015/10/14 Javascript
javascript实现tab切换特效
2015/11/12 Javascript
基于jQuery Tipso插件实现消息提示框特效
2016/03/16 Javascript
jquery结合html实现中英文页面切换
2016/11/29 Javascript
JavaScript的for循环中嵌套一个点击事件的问题解决
2017/03/03 Javascript
利用JavaScript在网页实现八数码启发式A*算法动画效果
2017/04/16 Javascript
Nodejs模块载入运行原理
2018/02/23 NodeJs
浅谈Vue Element中Select下拉框选取值的问题
2018/03/01 Javascript
vue移动端实现下拉刷新
2018/04/22 Javascript
vue实现打印功能的两种方法
2018/09/07 Javascript
详解微信小程序中组件通讯
2018/10/30 Javascript
vue 指令和过滤器的基本使用(品牌管理案例)
2019/11/04 Javascript
Python字符遍历的艺术
2008/09/06 Python
Python写的英文字符大小写转换代码示例
2015/03/06 Python
使用Protocol Buffers的C语言拓展提速Python程序的示例
2015/04/16 Python
构建Python包的五个简单准则简介
2015/06/15 Python
python简单猜数游戏实例
2015/07/09 Python
解决python3中cv2读取中文路径的问题
2018/12/05 Python
Django实现一对多表模型的跨表查询方法
2018/12/18 Python
python语言元素知识点详解
2019/05/15 Python
Python脚本破解压缩文件口令实例教程(zipfile)
2020/06/14 Python
城市观光通行证:The Sightseeing Pass
2018/04/28 全球购物
教师党性分析材料
2014/02/04 职场文书
司仪主持词两篇
2014/03/22 职场文书
教师节活动总结
2014/08/29 职场文书
运动会加油稿20字
2014/11/15 职场文书
婚前保证书范文
2015/02/28 职场文书
音乐课外活动总结
2015/05/09 职场文书
医院中层管理人员培训心得体会
2016/01/11 职场文书
2019垃圾分类宣传口号汇总
2019/08/16 职场文书
PHP控制循环操作的时间
2021/04/01 PHP
MySQL中datetime时间字段的四舍五入操作
2021/10/05 MySQL