tensorflow 设置显存占用自增长和指定GPU
2018-07-02 09:28 JohnRain 阅读(2857) 评论(0) 编辑 收藏 举报指定GPU
如果要在python代码中设置使用的GPU(如使用pycharm进行调试时),可以使用下面的代码
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "2"
制定显存
定量设置显存
默认tensorflow是使用GPU尽可能多的显存。可以通过下面的方式,来设置使用的GPU显存:
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.7)
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))
按需设置显存
gpu_options = tf.GPUOptions(allow_growth=True)
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))