如何使用tensorflow-gpu GPUOptions

Question

我正在尝试使用tensorflow-gpu，以便可以更快地运行模型，但是我不太了解如何激活它。目前，我的代码如下：

import tensorflow as tf

gpu_options = tf.compat.v1.GPUOptions(set_per_process_memory_fraction(0.333))
sess = tf.Session(config=tf.compat.v1.ConfigProto(gpu_options=gpu_options))

但是此代码显然不起作用，我是如何使用它。我已经看过tensorflow官方网站的文档，但这确实令人困惑。 有人可以告诉我如何正确使用它吗？

Answer 1

我假设您正在运行tensorflow 2.0版。在这种情况下，您可以在执行模式下设置GPU选项，如下所示：

from tensorflow.compat.v1 import ConfigProto
from tensorflow.compat.v1 import InteractiveSession

config = ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.333
session = InteractiveSession(config=config)

# model code as usual down here

我在GitHub修复RTX 2000系列CUDNN_STATUS_INTERNAL_ERROR GPU错误中找到了这个示例。