尝试为Tensorflow使用GPU时找不到各种CUDA dll文件



在尝试将NVIDIA GPU计算工具包与Tensorflow/Anaconda一起使用时,我遇到了CUDA安装后丢失各种dll文件的问题。这就是TF抱怨的:

2021-06-13 15:57:48.349493: W tensorflow/stream_executor/platform/default/dso_loader.cc:60] Could not load dynamic library 'cublas64_11.dll'; dlerror: cublas64_11.dll not found
2021-06-13 15:57:48.349942: W tensorflow/stream_executor/platform/default/dso_loader.cc:60] Could not load dynamic library 'cublasLt64_11.dll'; dlerror: cublasLt64_11.dll not found
2021-06-13 15:57:48.350349: W tensorflow/stream_executor/platform/default/dso_loader.cc:60] Could not load dynamic library 'cufft64_10.dll'; dlerror: cufft64_10.dll not found
2021-06-13 15:57:48.350752: W tensorflow/stream_executor/platform/default/dso_loader.cc:60] Could not load dynamic library 'curand64_10.dll'; dlerror: curand64_10.dll not found
2021-06-13 15:57:48.351155: W tensorflow/stream_executor/platform/default/dso_loader.cc:60] Could not load dynamic library 'cusolver64_10.dll'; dlerror: cusolver64_10.dll not found
2021-06-13 15:57:48.351555: W tensorflow/stream_executor/platform/default/dso_loader.cc:60] Could not load dynamic library 'cusparse64_11.dll'; dlerror: cusparse64_11.dll not found

最初,cudart64_110.dll也不见了,但我手动下载了它,并将其放入C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3\bin中。这不是环境变量(即路径(的问题——安装CUDA后,无论是在上述工具包文件夹中,还是在其他任何地方,上述DLL都不可用。我也不能简单地安装CUDA的旧版本,因为安装工具提到我已经安装了新版本。

在这些DLL不可用的情况下,我缺少什么?

TF需要CUDA 11.0,正如评论中善意指出的那样。奇怪的是,我的RTX 3070的性能明显比我的i9-10850K差(训练一个批量大小为128的密集序列模型(,但这是我需要研究的另一个主题。

最新更新