我正在尝试建立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才能正常工作吗?任何帮助都将不胜感激。谢谢!
发布于 2019-08-16 17:50:38
据我所知,PlaidML没有工作,因为我没有必要的先决条件,如OpenCL。一旦我下载了Visual C++构建工具,就可以从PyopenCL文件中安装.whl。这似乎解决了这个问题。
https://stackoverflow.com/questions/57504746
复制相似问题