我使用的是 Tensorflow-gpu 1.10,因为我在使用 cuDNN 时遇到了问题。我想将它使用的 gpu 限制在 40%,所以我做到了
config.gpu_options.per_process_gpu_memory_fraction = 0.4
session = tf.Session(config=config)
我查看了 Omen Control Center 以查看我使用了多少 GPU,而它使用了大约 70% 的 GPU。我没有收到任何错误消息。我的 GPU 是 GTX 1080。我在网上找遍了其他人也有类似的问题,但没有一个解决方案有效。我也将 Keras 用于模型和图层。我也在 Jupyter Notebook 中这样做。
最佳答案
试试这个:
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction = 0.4)
session = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options)
关于python - Tensorflow per_process_gpu_memory_fraction 不工作,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/54174740/