如何通过 TensorFlow 强制 Keras 在 R 中使用 GPU

【问题标题】：How to force Keras with TensorFlow to use the GPU in R如何通过 TensorFlow 强制 Keras 在 R 中使用 GPU
【发布时间】：2018-09-03 05:45:35
【问题描述】：

我按照 R 的 Tensorflow 和 Keras 安装说明进行操作。我有 Keras（Ubuntu 16.04 上的 python3），但它拒绝在我的 GPU 上运行。我尝试使用install_keras(tensorflow = "gpu") 卸载并重新安装，但它仍然只能在 CPU 上运行。

无论如何我可以检查配置以查看它是否配置为在 CPU 或 GPU 上运行？

我可以强制它在 GPU 上运行吗？

【问题讨论】：

您是否安装了cuda 和cudnn？

标签： r tensorflow keras

【解决方案1】：

您的 tensorflow GPU 安装失败，因为否则 keras 会默认尝试在 GPU 上启动计算。您必须完全遵循 tf 的 GPU 安装说明，并且需要准确的 CUDA 和 cuDNN 版本：

检查您是否安装了 CUDA 9.0 和 cuDNN > 7.2 (follow the install instructions here)
用install_keras(tensorflow = "gpu")重新安装keras
您可以列出设备：

library(keras) k = backend() sess = k$get_session() sess$list_devices()

【讨论】：