安装TensorFlow GPU版后导入报错:找不到cudart64_90.dll
解决TensorFlow GPU导入时找不到
cudart64_90.dll的问题 从你的错误日志能一眼揪出核心问题:你安装的CUDA 9.1和当前TensorFlow GPU版本依赖的CUDA版本不匹配。TensorFlow启动时会检查对应版本的CUDA运行库,而CUDA 9.1里只有cudart64_91.dll,自然找不到它要求的cudart64_90.dll。
下面给你两个实用的解决方案,你可以根据自己的需求选择:
方案一:降级CUDA到9.0版本
这是最贴合当前TensorFlow版本的做法:
- 先卸载已安装的CUDA 9.1,下载并安装CUDA 9.0版本
- 同时要下载与CUDA 9.0严格对应的cuDNN版本,解压后把cuDNN里的
bin、include、lib三个文件夹内容,分别复制到CUDA安装目录的对应文件夹中 - 确认CUDA的
bin路径已添加到系统%PATH%环境变量,之后重启终端或IDE,再尝试导入TensorFlow
方案二:升级TensorFlow到支持CUDA 9.1的版本
如果你不想降级CUDA,可以选择升级TensorFlow适配现有环境:
- 先卸载当前TensorFlow GPU版本:
pip uninstall tensorflow-gpu - 安装支持CUDA 9.1的TensorFlow GPU版本(比如TensorFlow 1.10及以后的版本大多支持CUDA 9.1),执行命令:
pip install tensorflow-gpu==<你需要的版本号> - 同样要确认cuDNN版本与CUDA 9.1匹配,环境变量配置正确,重启开发环境后再测试导入
最后补充两个容易忽略的检查点:
- 安装完CUDA和cuDNN后,必须重启你的开发工具(比如PyCharm、CMD等),否则环境变量的修改不会生效
- 确认Python 3.6.5符合你选择的TensorFlow版本要求(大部分TensorFlow 1.x版本都兼容Python 3.6)
内容的提问来源于stack exchange,提问作者Rahul
相关产品推荐
相关产品推荐

