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

Posted 2023-02-23

【中文标题】在 GPU 上激活 Tensorflow 2.0 时遇到问题

【英文标题】：Trouble activating Tensorflow 2.0 on GPU

【发布时间】：2020-07-08 22:07:03

【问题描述】：

这里已经被问过无数次了，但我找不到适合我的问题的解决方案。 我正在尝试在 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。我应该怎么做才能解决这个问题？

【问题讨论】：

这是你可以参考的。这可能是因为你的 CUDA 被破坏了。 Cudart64-101-dll

【参考方案1】：

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

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

希望这会有所帮助。

【讨论】：

我现在正在尝试下载 CUDA 10.1，但我收到来自 NVIDIA 的安装错误。我看看能做些什么

当我尝试重新安装 10.1 版本时，显然 CUDA 10.2 导致了一些问题。我手动下载并只安装了 CUDA（没有安装 CUDA 时通常会出现的驱动程序等东西）并且它有效