TensorFlow 强制使用 GPU

【问题标题】：TensorFlow mandatory GPU useTensorFlow 强制使用 GPU
【发布时间】：2018-03-27 00:05:08
【问题描述】：

我们的 GPU 处于独占模式。有时某些用户可能会手动登录机器并窃取 GPU。

当 GPU 初始化在 TensorFlow 脚本中失败时，如何引发异常？我注意到，当 TensorFlow 无法初始化 GPU 时，它会打印出一条错误消息，但会在 CPU 上运行。我想停止它而不是在 CPU 上运行。

【问题讨论】：

【解决方案1】：

如果您使用以下命令强制图形的任何部分在 GPU 上运行：

with tf.device('/device:GPU:0'):

然后您的会话变量初始化程序将停止运行并在没有可用 GPU 时抛出 InvalidArgumentError。

【讨论】：