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

Question

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

有没有办法检查配置以查看它是否配置为在CPU或GPU上运行？

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

- Gabriel Fair

---

你是否已经安装了 cuda 和 cudnn？ - Milo Lu

Answer 1

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

1. 检查是否已安装CUDA 9.0和cuDNN>7.2（请按此处的安装说明操作）
2. 使用install_keras(tensorflow="gpu")重新安装keras

3. 您可以使用以下命令列出设备：

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

- Max