张小白教你安装Windows 11的GPU驱动(CUDA 11.1和cuDNN 8.6.0 for CUDA 11.X)
两年前,张小白为了安装某个国产AI框架,还特意装过Windows上的CUDA 10.0和cuDNN 7.5.0 for CUDA 10.0:https://bbs.huaweicloud.cn/blogs/212446
后来玩转另一个国产AI框架,也就是我们华为的昇思MindSpore,只是它的Windows GPU版本总是不见踪影,所以安装的驱动貌似也没起到什么作用,直到最近MindSpore for Windows版本才浮出水面:https://zhuanlan.zhihu.com/p/574507930
仔细看了它的安装要求,它需要安装CUDA 11.1/11.6,以及对应的cuDNN 8.6.0 for CUDA 11.X的版本,然而两年过去了,这些软件的安装发生了变化了吗?我们并不知道,那就尝试一下吧!
一、安装CUDA 11.1 for Windows
1、下载CUDA 11.1 for Windows安装包
打开CUDA下载页面:https://developer.nvidia.com/cuda-11.1.1-download-archive
选择Windows local版本下载,具体链接为:https://developer.download.nvidia.cn/compute/cuda/11.1.1/local_installers/cuda_11.1.1_456.81_win10.exe
耐心等待下载完毕。
2、安装CUDA 11.1 for Windows
双击cuda开头的exe文件:
点击OK
等待安装程序加载完毕:
点击继续:
同意:
选择自定义后,点击下一步:
选择全部安装,点击下一步:
暂不修改路径,点击下一步:
耐心等待安装过程:
这里提示需要修改PATH环境变量:PATH需加上:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\libnvvp;C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\bin;
我们可以从 控制面板中的编辑环境变量菜单进行增加:
出现以下界面:
说明基本安装完毕:
这里会提示有些装了,有些没装,暂时不管这些。点击关闭按钮。
这里需要再添加 CUDA_PATH这个环境变量:
我们把现在的10.0改为11.1
重启下Windows。
这就暂时完成了CUDA 11.1 for windows的安装。
二、安装cuDNN 8.6.0 for CUDA 11.X
1、下载cuDNN 8.6.0 for CUDA 11.X for Windows安装包
打开cuDNN下载页面:https://developer.nvidia.com/cudnn
此时需要登陆,如果没有注册过的人需要点击Join Now,张小白直接点击Login:
输入用户名,密码,点击登录:
系统会发个邮件给你,打开自己的邮箱:
查看邮件,点击邮件中的 Verify Email Address:
邮件验证成功!
原来的登录页面会自动跳转登录:
不一会儿,就打开了下载页面:
选择cuDNN 8.6.0 for CUDA 11.x windows版本即可。
耐心等待软件下载完毕。
2、安装cuDNN 8.6.0 for CUDA 11.X
下面开始安装cuDNN:
按张小白以往安装cuDNN for CUDA 10的经验,一般来说是将解压后的cuDNN包拷贝到 CUDA的include,lib,bin等目录中去。
先解压:
再分别拷贝:
include目录:
bin目录:
lib下的x64目录:
增加一个CUDNN_HOME的环境变量:
此时,cuDNN应该是安装完毕了吧。。。
三、安装验证
检查CUDA版本:
在cmd提示符下执行
nvcc -V
可以看出是11.1版本。
cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\extras\demo_suite
执行 deviceQuery.exe
可以看到CUDA的驱动是11.6版本,Runtime是11.1版本。
执行bandwidthTest.exe
以上显示表示CUDA和cuDNN都安装成功。
打开 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\include\cudnn_version.h
可以看到cuDNN的版本为 8.6.0(主版本-小版本-补丁版本)
(全文完,谢谢阅读)
- 点赞
- 收藏
- 关注作者
评论(0)