在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。

希望这会有所帮助。