- 博客(1)
- 收藏
- 关注
原创 Tensorflow设置gpu显存使用比例
一般情况下,tf程序会再运行时占用gpu所有显存。可以通过会话设置,使得代码只是用gpu一定比例的显存: config = tf.ConfigProto() congfig.gpu_options.per_process_gpu_memory_fraction = 0.4 sess = tf.Session(config =config ) ...
2018-10-22 16:05:47 1297
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人