安装keras tensorflow指定版本后需要安装指定版本的CUDA和cudnn,否则报错 ImportError: Could not find ‘cudart64_90.dll’.
解决方案
下载cuda9.0.176,但是这个在官网下载会发现这东西没有。
这里我给出下载的网盘地址百度网盘
下载完后安装即可。
一般默认的位置安装的位置在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
下载cudnn7.0.5
下载地址
下载 cuDNN 时需要注册英伟达的账号,建议使用谷歌账号登录或 gmail 邮箱注册,我选择谷歌账号登录,方便快捷。
下载完解压后,把将解压后的三个文件夹 bin,include,lib 复制到 CUDA 的安装路径下和重名的三个文件夹进行合并。CUDA安装目录默认是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0