一般情况下,tf程序运行时会占用gpu所有显存。可以通过会话设置,使得代码只是用gpu一定比例的显存:
config = tf.ConfigProto()
congfig.gpu_options.per_process_gpu_memory_fraction = 0.4
sess = tf.Session(config =config )
Tensorflow设置gpu显存使用比例
最新推荐文章于 2024-07-13 01:33:51 发布
一般情况下,tf程序运行时会占用gpu所有显存。可以通过会话设置,使得代码只是用gpu一定比例的显存:
config = tf.ConfigProto()
congfig.gpu_options.per_process_gpu_memory_fraction = 0.4
sess = tf.Session(config =config )