tensorflow：在一个GPU上训练网络但占用另一个GPU的空间

Question

我有两个GPU，我只想使用一个GPU通过tensorflow训练网络。当我训练它时，代码使用两个GPU的所有记忆，但只有一个GPU正在工作：

我不知道为什么以及如何解决这个问题。

Answer 1

尝试设置：

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

，如question。

Answer 2

默认情况下，tensorflow将占用所有可用GPU的内存。您可以根据@ sygi的答案设置允许增长，或者根据Yaroslav的评论，只对其中一个GPU显示tensorflow。并按照this question。