根据Tensorflow-GPU安装CUDA+CUDNN
背景:
win10要安装TensorFlow-GPU 1.4.0,现有CUDA为10.0,版本不符
问题:
ImportError:
Could not find 'cudart64_80.dll'.
TensorFlow requires that this DLL be installed in a directory that is named in your %PATH% environment variable.
Download and install CUDA 8.0 from this URL: https://developer.nvidia.com/cuda-toolkit
读题,是TensorFlow-GPU和CUDA版本不符,要CUDA8.0。
ImportError:
Could not find 'cudnn64_6.dll'.
TensorFlow requires that this DLL be installed in a directory that is named in your %PATH% environment variable.
Note that installing cuDNN is a separate step from installing CUDA,
and this DLL is often found in a different directory from the CUDA DLLs.
You may install the necessary DLL by downloading cuDNN 6 from this URL: https://developer.nvidia.com/cudnn
读题,是TensorFlow-GPU和CUDNN版本不符,要CUDNN6.0。
解决:
1 版本关系
要安装TensorFlow-GPU,要注意TF-GPU和CUDA、CUDNN的版本关系:(其实仔细看问题,就可以知道要下什么版本的了)
CUDA现有版本10.1,所以要去官网重装(版本太高不行的原因,可能是锅配盖,合适就行)。
2 如何查看现有CUDA版本号
推荐几个方法
Windows系统查看CUDA版本号(也可以在右下角找到NVDIA的标志,右键即可)
3 下载安装CUDA
3.1 下载
如果仔细看问题1,你就可以看到后面附赠一个URL,很贴心。
附赠一个官网下载链接:CUDA Toolkit Archive,点进去,找合适自己的版本就可以了。
根据自己的系统选择,没毛病。
Installer Type安装类型可以选择网络安装还是本地安装,个人建议本地。
3.2 安装
自定义推荐都可以吧,忍不住点个自定义瞅瞅
安装完毕,检测下版本号
在cmd也可以
4 安装下载CUDNN
介绍:深度神经网络库(cuDNN)是GPU加速的用于深度神经网络的原语库。cuDNN为标准例程提供了高度优化的实现,例如向前和向后卷积,池化,规范化和**层。
如果能打开这个页面:cuDNN Archive,下载对应版本,可以跳过4.1\4.2,从4.3开始。(能少走点弯路就少走一点)
4.1 前置步骤
又可以在问题2中找到官网URL:NVIDIA cuDNN
有点麻烦,他要求注册登录,还要写调查问卷。
(耐心点,可以用微信、QQ,但还是要绑定用户)
邮箱验证
填写细节
居然还有调查问卷!写写写
4.2 下载
终于到了下载页面,先打勾✔,发现版本不是我要的,点最下面的红框里的内容。
网址:https://developer.nvidia.com/rdp/cudnn-download
https://developer.nvidia.com/rdp/cudnn-archive
选择合适版本下载
4.3 安装
先打开之前安装的cuda的文件,默认位置是在 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA
如同,将CUDNN 压缩包中的文件,移动到相同文件名下。
4.4 环境配置
参考了三篇博文就有三种配置,先选了最简单的试试
往path里面加一条(注意版本号)
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64
最简单的就搞定了。
参考
查看版本号
Tensorflow不同版本要求与CUDA及CUDNN版本对应关系
tensorflow-gpu版本与CUDA,Cudnn兼容问题,及GPU 查阅技巧
CUDA+CUDNN的教程
上一篇: Flutter TabBar、TabBarView的使用
下一篇: 《Flutter 控件大全》第二十六个:ConstrainedBox、UnconstrainedBox、SizedBox、AspectRatio
推荐阅读
-
tensorflow-gpu安装的常见问题及解决方案
-
记录:Ubuntu 18.04 安装 tensorflow-gpu 版本
-
详细完整的Tensorflow-gpu环境安装教程
-
tensorflow-gpu安装的常见问题及解决方案
-
CUDA9.1上安装tensorflow-gpu的过程
-
记录:Ubuntu 18.04 安装 tensorflow-gpu 版本
-
ubuntu18.40 rtx2080ti安装显卡驱动/cuda/cudnn/tensorflow-gpu
-
W7,显卡型号nvidia geforce 840M,安装tensorflow-gpu
-
百度AI studio安装Tensorflow-gpu 2.0.0
-
tensorflow-gpu的安装