设置Keras和TensoFlow以使用AMD GPU运行

y0u0uwnf  于 2023-10-19  发布在  其他
关注(0)|答案(1)|浏览(107)

我正在尝试设置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正常工作?

mzmfm0qo

mzmfm0qo1#

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

相关问题