基于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使用CMD模块更优雅的运行脚本
May 11 Python
python实现在字符串中查找子字符串的方法
Jul 11 Python
Python 3中的yield from语法详解
Jan 18 Python
详解用python实现简单的遗传算法
Jan 02 Python
python 按不同维度求和,最值,均值的实例
Jun 28 Python
Selenium chrome配置代理Python版的方法
Nov 29 Python
Python3+Appium实现多台移动设备操作的方法
Jul 05 Python
tf.concat中axis的含义与使用详解
Feb 07 Python
Python基础之字典常见操作经典实例详解
Feb 26 Python
Python之关于类变量的两种赋值区别详解
Mar 12 Python
基于python实现计算两组数据P值
Jul 10 Python
python 实现数据库中数据添加、查询与更新的示例代码
Dec 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代码把全角数字转为半角数字
2007/12/10 PHP
PHP通用检测函数集合
2011/02/08 PHP
php多文件上传功能实现原理及代码
2013/04/18 PHP
解析在PHP中使用mysqli扩展库对mysql的操作
2013/07/03 PHP
ThinkPHP采用GET方式获取中文参数查询无结果的解决方法
2014/06/26 PHP
简单PHP会话(session)说明介绍
2016/08/21 PHP
PHP快速排序quicksort实例详解
2016/09/28 PHP
php从数据库中读取特定的行(实例)
2017/06/02 PHP
基于Jquery制作的幻灯片图集效果打包下载
2011/02/12 Javascript
情人节专属 纯js脚本1k大小的3D玫瑰效果
2012/02/11 Javascript
js中的replace方法使用介绍
2013/10/28 Javascript
jQuery实现Twitter的自动文字补齐特效
2014/11/28 Javascript
整理AngularJS中的一些常用指令
2015/06/16 Javascript
js+css实现的圆角边框TAB选项卡滑动门代码分享(2款)
2015/08/26 Javascript
浅析JS运动
2015/12/28 Javascript
jQuery头像裁剪工具jcrop用法实例(附演示与demo源码下载)
2016/01/22 Javascript
js中数组的常用方法小结
2016/12/30 Javascript
vue component组件使用方法详解
2017/07/14 Javascript
Vue.js 中制作自定义选择组件的代码附演示demo
2020/02/28 Javascript
IDEA配置jQuery, $符号不再显示黄色波浪线的问题
2020/10/09 jQuery
python遍历类中所有成员的方法
2015/03/18 Python
python中类和实例如何绑定属性与方法示例详解
2017/08/18 Python
Python错误处理操作示例
2018/07/18 Python
python实现的发邮件功能示例
2019/09/11 Python
python tkinter GUI绘制,以及点击更新显示图片代码
2020/03/14 Python
python3排序的实例方法
2020/10/20 Python
Burberry英国官网:英国标志性奢侈品牌
2017/03/29 全球购物
BrandAlley英国:法国折扣奢侈品网上零售商
2017/07/03 全球购物
市场部专员岗位职责
2013/11/30 职场文书
青蓝工程实施方案
2014/03/27 职场文书
怎样填写就业意向
2014/04/02 职场文书
文明礼貌演讲稿
2014/05/12 职场文书
大学生考试作弊检讨书1000字
2014/10/14 职场文书
护士个人年终总结
2015/02/13 职场文书
2016年村干部公开承诺书(公开承诺事项)
2016/03/25 职场文书
你知道哪几种MYSQL的连接查询
2021/06/03 MySQL