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

时间：2018-09-03 05:45:35

标签： r tensorflow keras

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

我是否仍然可以检查配置以查看其是否配置为在CPU或GPU上运行？

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

enter image description here enter image description here enter image description here

1 个答案:

答案 0 :(得分：2)

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

  1. 检查是否已安装CUDA 9.0和cuDNN> 7.2（follow the install instructions here）
  2. 使用install_keras(tensorflow = "gpu")重新安装keras

  3. 您可以使用以下方式列出设备：

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

相关问题
最新问题