如何使用 AMD 显卡导入 tensorflow-GPU，或任何解决方法

Question

我有 AMD 显卡。现在，当我尝试导入 tensorflow 时，出现以下错误。 我认为（据我所知）如果我使用普通的 TF 而不是 TF-GPU，那么这个问题可能会解决。但是，我确实想使用显卡，因为否则我的强化学习算法真的很慢。有什么解决办法吗？

ImportError：找不到“nvcuda.dll”。 TensorFlow 要求 此 DLL 安装在您的 %PATH% 中命名的目录中 环境变量。通常它安装在 'C:\Windows\System32'。如果它不存在，请确保您有一个 安装了正确驱动程序的支持 CUDA 的 GPU。

Answer 1

目前，对于 AMD，您可以通过两种方式进行操作。 要么使用最新的 AMD 的 ROCm 安装 tensorflow。

official ROCm install

和

official ROCm tensorflow install

检查您是否使用支持的 AMD GPU 检查here

如果您的显卡不支持 ROCm，则使用 TensorFlow 的 OpenCL 实现

https://github.com/benoitsteiner/tensorflow-opencl

或

https://github.com/hughperkins/tf-coriander