美文网首页深度学习的各种骚套路机器学习
tensorflow 设置显存占用自增长和指定GPU

tensorflow 设置显存占用自增长和指定GPU

作者: 大海龟啦啦啦 | 来源:发表于2020-06-21 22:25 被阅读0次

指定GPU

如果要在python代码中设置使用的GPU(如使用pycharm进行调试时),可以使用下面的代码

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

制定显存

1、定量设置显存

默认tensorflow是使用GPU尽可能多的显存。可以通过下面的方式,来设置使用的GPU显存

gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.7)
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))

2、按需设置显存

gpu_options = tf.GPUOptions(allow_growth=True)
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))

相关文章

网友评论

    本文标题:tensorflow 设置显存占用自增长和指定GPU

    本文链接:https://www.haomeiwen.com/subject/kvyzxktx.html