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

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

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

标签： tensorflowgpu

目前，现在使用 AMD，您可以通过两种方式来实现它。要么使用最新的 AMD 的 ROCm 来安装 tensorflow。

和

检查您是否使用支持的 AMD GPU 在此处检查

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

或者