设置 keras 和 tensorflow 以与 AMD GPU 一起运行

Setting up keras and tensoflow to operate with AMD GPU

我正在尝试设置 Keras 以便 运行 使用我的 GPU 建模。我有一个 Radeon RX580 并且 运行ning Windows 10.

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

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