您当前的位置:首页 > 计算机 > 编程开发 > Python

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

时间:06-26来源:作者:点击数:

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

1.pip install tensorflow-gpu下载过慢

解决办法可查看 Python机器学习常用模块

2.安装完tensorflow以后,运行时出现ImportError: Could not find ‘cudart64_90.dll’

解决办法:在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 上注册,然后下载相应版本,解压缩文件,然后将D:\cudnn-9.2-windows10-x64-v7.1\cuda\bin\cudnn64_7.dll 文件拷贝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin里面,就大功告成了

方便获取更多学习、工作、生活信息请关注本站微信公众号城东书院 微信服务号城东书院 微信订阅号
推荐内容
相关内容
栏目更新
栏目热门