我只想知道是否有办法避免im2txt使用GeForce 970 GTX的超低0.5GB。我们可以强制训练课程只使用常规的3,5GB吗？这是个好主意吗？
感谢。
答案 0 :(得分：1)
您可以尝试使用以下命令限制内存总量。不确定分配哪个内存，如果你运气不好可能包括慢速0.5GB
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.7)
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))