我正在尝试设置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正常工作?
1条答案
按热度按时间mzmfm0qo1#
据我所知,PlaidML没有工作,因为我没有必要的先决条件,如OpenCL。有一次,我下载了Visual Studio C++构建工具,以便从.whl文件安装PyopenCL。这似乎解决了问题