Tensorflow per_process_gpu_memory_fraction不起作用

Question

我正在使用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是GTX1080。我在网上看了很多，其他人也遇到了类似的问题，但是没有一个解决方案有效。我还将Keras用于模型和图层。我也在Jupyter Notebook中执行此操作。

Answer 1

尝试这个：

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