我试图设置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上运行,并且即使plaidml设置成功也没有GPU。我已经读到需要PyOpenCl,但是对于为什么或什么容量还没有明确的答案。有人知道如何设置此AMD GPU才能正常工作吗?任何帮助将非常感激。谢谢!
答案 0 :(得分:0)
据我所知,PlaidML无法正常工作,因为我没有必需的先决条件,例如OpenCL。一旦我下载了Visual Studio C ++构建工具,以便从.whl文件安装PyopenCL。这似乎解决了问题