【发布时间】: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