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

Question

我遵循了 R 的 Tensorflow 和 Keras 安装说明。我有 Keras（Ubuntu 16.04 上的 python3），但它拒绝在我的 GPU 上运行。我试过只是卸载并重新安装install_keras(tensorflow = "gpu")，它仍然只能在 CPU 上运行。

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

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

Answer 1

您的 tensorflow GPU 安装失败，否则 keras 会尝试在 GPU 上默认启动计算。您必须完全按照 tf 的 GPU 安装说明进行操作，并且需要确切的 CUDA 和 cuDNN 版本：

1. 检查您是否安装了 CUDA 9.0 和 cuDNN > 7.2（按照此处的安装说明进行操作）
2. 重新安装keras install_keras(tensorflow = "gpu")

3. 您可以列出设备：

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