(转)tensorflow中使用指定的GPU及GPU显存
来源:互联网 发布:智人 美洲 知乎 编辑:程序博客网 时间:2024/05/28 23:10
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
转载请注明出处:
http://www.cnblogs.com/darkknightzh/p/6591923.html
参考网址:
http://stackoverflow.com/questions/36668467/change-default-gpu-in-tensorflow
http://stackoverflow.com/questions/37893755/tensorflow-set-cuda-visible-devices-within-jupyter
回到顶部(go to top)
1 终端执行程序时设置使用的GPU
如果电脑有多个GPU,tensorflow默认全部使用。如果想只使用部分GPU,可以设置CUDA_VISIBLE_DEVICES。在调用python程序时,可以使用(见第一个参考网址Franck Dernoncourt的回复):
CUDA_VISIBLE_DEVICES=1 python my_script.py
Environment Variable Syntax ResultsCUDA_VISIBLE_DEVICES=1 Only device 1 will be seenCUDA_VISIBLE_DEVICES=0,1 Devices 0 and 1 will be visibleCUDA_VISIBLE_DEVICES="0,1" Same as above, quotation marks are optionalCUDA_VISIBLE_DEVICES=0,2,3 Devices 0, 2, 3 will be visible; device 1 is maskedCUDA_VISIBLE_DEVICES="" No GPU will be visible
回到顶部(go to top)
2 python代码中设置使用的GPU
如果要在python代码中设置使用的GPU(如使用pycharm进行调试时),可以使用下面的代码(见第二个参考网址中Yaroslav Bulatov的回复):
import osos.environ["CUDA_VISIBLE_DEVICES"] = "2"
回到顶部(go to top)
3 设置tensorflow使用的显存大小
默认tensorflow是使用GPU尽可能多的显存。可以通过下面的方式,来设置使用的GPU显存:
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.7)sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))
上面分配给tensorflow的GPU显存大小为:GPU实际显存*0.7。
可以按照需要,设置不同的值,来分配显存。
阅读全文
1 0
- tensorflow中使用指定的GPU及GPU显存
- (转)tensorflow中使用指定的GPU及GPU显存
- TensorFlow指定特定GPU以及占用显存的比例
- tensorflow指定gpu的使用
- tensorflow GPU显存控制
- tensorflow使用GPU训练时的显存占用问题
- tensorflow使用GPU训练时的显存占用问题
- tensorflow使用GPU训练时的显存占用问题
- tensorflow中指定GPU及GPU显存设置
- Tensorflow指定使用哪个GPU
- 指定TensorFlow使用哪一个GPU
- 【Ubuntu-Tensorflow】GPU设置及显存资源分配
- TensorFlow使用GPU训练网络时多块显卡的显存使用问题
- Keras限制GPU显存使用
- Tensorflow 多GPU下,指定使用某一块GPU
- tensorflow 指定GPU
- Tensorflow 指定GPU运行
- 17、TensorFLow GPU 的使用
- 常用字符集总结
- 如何让努力变得更加有效
- mybatis 一对多映射中的一些小问题
- thinkphp3.2加入验证码
- Terminal信息导出到Excel模板--【sql+entity+dao丶Service+Controller】
- (转)tensorflow中使用指定的GPU及GPU显存
- 斯坦福机器学习 ex1 Python实现
- opencv如何读取文件夹中的图片
- 安装使用Drill-1.10
- 分治——棋盘覆盖
- Cx_Redis深度学习(一)--发布订阅
- 44、翻转单词顺序列
- MapReduce On YARN的工作原理
- 深入浅出WPF