如何使用 tensorflow-gpu GPUOptions答案

【问题标题】：How to use tensorflow-gpu GPUOptions如何使用 tensorflow-gpu GPUOptions
【发布时间】：2019-10-23 20:06:17
【问题描述】：

我正在尝试使用 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 网站文档，但这确实令人困惑。谁能告诉我如何正确使用它？

【问题讨论】：

标签： tensorflow

【解决方案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 错误时找到了这个示例。

【讨论】：