张量流 per_process_gpu_memory_fraction 不起作用答案

【问题标题】：Tensorflow per_process_gpu_memory_fraction not working张量流 per_process_gpu_memory_fraction 不起作用
【发布时间】：2019-01-14 01:13:46
【问题描述】：

我正在使用 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 中这样做。

【问题讨论】：

标签： python python-3.x tensorflow keras

【解决方案1】：

试试这个：

gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction = 0.4)
session = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options)

【讨论】：