TensorFlow gpu版内存利用问题

Question

似乎默认是只使用显存不用内存的，那么有什么办法利用内存么？

Answer 1

tensorflow默认优先使用gpu内存，可以通过设置gpu内存的使用分配比例。

gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.4)

sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))

如上，如果gpu内存是10G的话，那么gpu内存最大会使用4G。
当tensorflow需要调用超过4G的内存计算时，则调用电脑内存。