gpt4 book ai didi

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

转载 作者:行者123 更新时间:2023-12-02 13:13:11 24 4
gpt4 key购买 nike

我按照 R 的 Tensorflow 和 Keras 安装说明进行操作。我有 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

最佳答案

您的 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. 您可以通过以下方式列出设备:

    库(keras)
    k = 后端()
    sess = k$get_session()
    sess$list_devices()

关于r - 如何强制 Keras with TensorFlow 在 R 中使用 GPU,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/52143756/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com