您的位置：首页 > 其它

Tensorflow 指定GPU运行

2017-10-13 10:13 183 查看
参考网址：http://blog.csdn.net/u014381600/article/details/72911262

1.终端执行程序时设置使用的GPU（linux终端）

CUDA_VISIBLE_DEVICES=1,2  python my_script.py

想指定几块GPU就标明GPU序号即可，中间用逗号隔开

2.python代码中设置使用的GPU（ipython或pycharm调试时）

import os 

os.environ["CUDA_VISIBLE_DEVICES"] = "2"

3.设置tensorflow使用的显存大小

gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.7) 

sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))

内容来自用户分享和网络整理，不保证内容的准确性，如有侵权内容，可联系管理员处理 点击这里给我发消息
标签： 
相关文章推荐
新的分享
章节导航