libing101007的博客

原创 Tensorflow设置gpu显存使用比例

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