彻底解决tensorflow:ImportError: Could not find 'cudart64_90.dll'

素颜马尾好姑娘i 2022-05-17 00:50 102阅读 0赞

今天装tensorflow-gpu出现了很多问题

1.pip install tensorflow-gpu下载过慢  
解决办法可查看 [Python机器学习常用模块][Python]

2.安装完tensorflow以后，运行时出现ImportError: Could not find ‘cudart64\_90.dll’  
解决办法：在[https://developer.nvidia.com/cuda-toolkit][https_developer.nvidia.com_cuda-toolkit]上下载CUDA 9.0,一定要下载对应的版本，然后安装，  
一般默认的位置安装的位置在C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v9.0。  
安装完以后，再设置环境变量  
CUDA\_PATH = C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v9.0  
CUDA\_PATH\_V9\_0 = C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v9.0  
CUDA\_SDK\_PATH = C:\\ProgramData\\NVIDIA Corporation\\CUDA Samples\\v9.0  
CUDA\_LIB\_PATH = %CUDA\_PATH%\\lib\\x64  
CUDA\_BIN\_PATH = %CUDA\_PATH%\\bin  
CUDA\_SDK\_BIN\_PATH = %CUDA\_SDK\_PATH%\\bin\\win64  
CUDA\_SDK\_LIB\_PATH = %CUDA\_SDK\_PATH%\\common\\lib\\x64

3.安装完CUDA 9.0以后，还是出现:ImportError: Could not find ‘cudnn64\_7.dll’  
解决办法:在[https://developer.nvidia.com/cudnn][https_developer.nvidia.com_cudnn] 上注册，然后下载相应版本，解压缩文件，然后将D:\\cudnn-9.2-windows10-x64-v7.1\\cuda\\bin\\cudnn64\_7.dll 文件拷贝到C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v9.0\\bin里面，就大功告成了

[Python]: https://blog.csdn.net/u011311291/article/details/77769283
[https_developer.nvidia.com_cuda-toolkit]: https://developer.nvidia.com/cuda-toolkit
[https_developer.nvidia.com_cudnn]: https://developer.nvidia.com/cudnn