在 GPU 上激活 Tensorflow 2.0 时遇到问题

Question

这里已经被问过无数次了，但我找不到适合我的问题的解决方案。 我正在尝试在 Windows 10 Anaconda 上安装 tensorflow-gpu。我下载了 CUDA（10.2 版）和 CuDNN，我已将文件从 CuDNN 库移动到 CUDA 文件夹，并将以下变量添加到 PATH：

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\libnvvp

在 anaconda 中运行以下代码后，我得到以下响应：

from tensorflow.python.client import device_lib print(device_lib.list_local_devices())

我猜主要问题是"Could not load dynamic library 'cudart64_101.dll'; dlerror: cudart64_101.dll not found"，所以我去了我的CUDA bin文件夹，没有cudart64_101.dll，但有cudart64_102.dll。我应该怎么做才能解决这个问题？

Answer 1

.dll 文件中的最后三个数字应该为您指明解决方案。 TensorFlow 正在寻找 CUDA 10.1，但您已经安装了 CUDA 10.2。

重新安装 CUDA 和兼容的 cuDNN 应该可以解决您的问题。有关 CUDA 和 cuDNN 的确切版本，请参阅 the TensorFlow installation tutorial。 在安装正确的 CUDA 库之后，您可能还必须重新安装 TF。

希望这会有所帮助。