Tensorflow设置gpu显存使用比例

版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/libing101007/article/details/83274277

一般情况下,tf程序运行时会占用gpu所有显存。可以通过会话设置,使得代码只是用gpu一定比例的显存:
config = tf.ConfigProto()
congfig.gpu_options.per_process_gpu_memory_fraction = 0.4
sess = tf.Session(config =config )

展开阅读全文

没有更多推荐了,返回首页