基于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编程时利用wxPython来支持多线程的方法
Apr 07 Python
Python与人工神经网络:使用神经网络识别手写图像介绍
Dec 19 Python
Python3实现的简单验证码识别功能示例
May 02 Python
分析python请求数据
Aug 19 Python
Python实现App自动签到领取积分功能
Sep 29 Python
python实现AES加密与解密
Mar 28 Python
python opencv 二值化 计算白色像素点的实例
Jul 03 Python
通过python实现弹窗广告拦截过程详解
Jul 10 Python
Python内存管理实例分析
Jul 10 Python
使用PyInstaller将Pygame库编写的小游戏程序打包为exe文件及出现问题解决方法
Sep 06 Python
Python-numpy实现灰度图像的分块和合并方式
Jan 09 Python
python连接PostgreSQL过程解析
Feb 09 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
ob_start(),ob_start('ob_gzhandler')使用
2006/12/25 PHP
php 随机生成10位字符代码
2009/03/26 PHP
php 全局变量范围分析
2009/08/07 PHP
yii框架配置默认controller和action示例
2014/04/30 PHP
实现png图片和png背景透明(支持多浏览器)的方法
2009/09/08 Javascript
JavaScript使用focus()设置焦点失败的解决方法
2014/09/03 Javascript
jquery自定义表单验证插件
2016/10/12 Javascript
进阶之初探nodeJS
2017/01/24 NodeJs
jQuery制作图片旋转效果
2017/02/02 Javascript
老生常谈jquery中detach()和remove()的区别
2017/03/02 Javascript
jQuery EasyUI tree增加搜索功能的实现方法
2017/04/27 jQuery
js判断数组是否包含某个字符串变量的实例
2017/11/24 Javascript
Vue传参一箩筐(页面、组件)
2019/04/04 Javascript
Layui 解决表格异步调用后台分页的问题
2019/10/26 Javascript
微信小程序自定义头部导航栏(组件化)
2019/11/15 Javascript
JavaScript设计模型Iterator实例解析
2020/01/22 Javascript
JS实现购物车基本功能
2020/11/08 Javascript
[01:34]2014DOTA2 TI预选赛预选赛 选手比赛房大揭秘!
2014/05/20 DOTA
python操作redis的方法
2015/07/07 Python
使用python 3实现发送邮件功能
2018/06/15 Python
Python分析彩票记录并预测中奖号码过程详解
2019/07/09 Python
django admin 自定义替换change页面模板的方法
2019/08/23 Python
python实现实时视频流播放代码实例
2020/01/11 Python
Django后台管理系统的图文使用教学
2020/01/20 Python
python matplotlib绘制三维图的示例
2020/09/24 Python
Sport-Thieme荷兰:购买体育用品
2019/08/25 全球购物
Mountain Hardwear官网:攀岩服装和户外装备
2019/09/26 全球购物
照片礼物和装饰:MyPhoto
2019/11/02 全球购物
五年级数学教学反思
2014/02/11 职场文书
八一建军节演讲稿
2014/09/10 职场文书
教师三严三实心得体会
2014/10/11 职场文书
获奖感言范文
2015/07/31 职场文书
感恩教师主题班会
2015/08/12 职场文书
导游词幽默开场白
2019/06/26 职场文书
解析:创业计划书和商业计划书二者之间到底有什么区别
2019/08/14 职场文书
win10壁纸在哪个文件夹 win10桌面背景图片文件位置分享
2022/08/05 数码科技