【问题标题】:Setting up keras and tensoflow to operate with AMD GPU设置 keras 和 tensorflow 以与 AMD GPU 一起运行
【发布时间】:2019-08-15 04:06:41
【问题描述】:

我正在尝试设置 Keras,以便使用我的 GPU 运行模型。我有一台 Radeon RX580 并且正在运行 Windows 10。

我发现 CUDA 仅支持 NVIDIA GPU,并且很难找到让我的代码在 GPU 上运行的方法。我尝试下载并设置 plaidml 但之后from tensorflow.python.client import device_lib print(device_lib.list_local_devices()) 只打印了我在 CPU 上运行并且没有可用的 GPU,即使 plaidml 设置成功。我已经读过 PyOpenCl 是必需的,但没有得到关于为什么或多大容量的明确答案。有谁知道如何设置这个 AMD GPU 才能正常工作?任何帮助将非常感激。谢谢!

【问题讨论】:

  • PlaidML 有一个 keras 后端,它与 TensorFlow AFAIK 无关。请详细分享您的实际操作。

标签: python-3.x tensorflow keras gpu amd


【解决方案1】:

据我所知,PlaidML 无法正常工作,因为我没有 OpenCL 等必需的先决条件。一旦我下载了 Visual Studio C++ 构建工具,以便从 .whl 文件安装 PyopenCL。这似乎解决了问题

【讨论】:

    猜你喜欢
    • 2016-10-19
    • 2020-08-11
    • 2019-01-14
    • 1970-01-01
    • 2021-06-04
    • 1970-01-01
    • 1970-01-01
    • 2021-10-03
    • 2018-12-30
    相关资源
    最近更新 更多