设置keras和tensoflow以与AMD GPU一起运行

时间:2019-08-15 04:06:41

标签: python-3.x tensorflow keras gpu amd

我试图设置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才能正常工作吗?任何帮助将非常感激。谢谢!

1 个答案:

答案 0 :(得分:0)

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