注意:本教程建立在您已经正确安装了显卡驱动的基础上
第一步,找到对应的CUDA版本
右键桌面,选择NVIDIA控制面板
控制面板是长这样的
然后我们点击控制面板左下角的系统信息,并选择组件
红框标出来的位置,就代表了你电脑上的显卡当前这个驱动最高支持的CUDA版本,比如我的电脑显卡现在最高支持的CUDA版本就是CUDA11.7,这意味着我可以装CUDA11.7及其更低的CUDA版本(cuda11.6、cuda11.1等)
第二步,查看你要安装的pytorch版本对应的cuda版本(如果只需要安装cuda可以跳过)
pytorch官网传送门:pytorch
我们点击Get Started
然后我们点击标框的这个链接,可以查看所有版本的pytorch
红框代表了这个版本pytorch支持的操作系统,然后黄框表示该版本pytorch支持的cuda版本
比如我目前打算要下载一个pytorch 1.10的版本
那么,我的cuda可以选择装10.2版本的 或者 装11.3版本的,我这里选择装cuda11.3 ,我们上面提到我们的显卡驱动目前最高可以支持cuda12.0,所以这里选择安装cuda11.3是完全没有问题呢,假如你的最新驱动最高能安装的cuda版本只有cuda11.0,那么你只能选择安装cuda10.2。
第二步,下载cuda
CUDA官方下载传送门:CUDA
如果打不开或者下载慢的可以从我另一篇文章上获取本地安装包:各版本CUDA和对应CUDNN离线安装包
然后我们在第一步决定安装cuda11.3,所以我们在官网找到这个cuda11.3的版本,然后左键点进去。然后按下图这样选择,最后点击下载
下载完后,我们开始安装
建议保持默认路径
如果这个系统上是第一次安装cuda就全选,如下图
如果之前已经安装过其它版本的cuda,就只保留第一项如下图
然后依旧保持默认路径,点击下一步
到这里cuda安装就完成了,然后我们打开命令窗口 输入nvcc -V,出下如下图所示的结果说明环境变量的配置也没有问题了
第三步,安装CUDNN加速库
cudnn官网下载传送门:cudnn,下载cudnn需要有NVIDIA的账号,可以自己注册
懒得注册的可以从我另一篇文章上获取本地安装包:各版本CUDA和对应CUDNN离线安装包
我们之前安装的cuda的版本是11.3的所以我们可以安装这个cudnn8.9.1因为它支持的cuda版本是11.X
然后我们下载这个windows版本的 zip压缩包
下载好后,我们将这个压缩包解压可以得到如下三个文件
我们将这三个文件夹复制,然后粘贴到cuda的根目录下,如果你没有修改默认路径的话,那就是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3
我这里已经是复制过去了,到此我们的整个安装过程就完成啦!!!!
windwos下安装cudnn是不需要配置环境变量的,如果担心自己的CUDA或者CUDNN安装不成功,可以按照我另一篇文章上的操作去验证是否安装成功。
链接 :windows下验证CUDA和CUDNN是否都安装成功
如果您发现该资源为电子书等存在侵权的资源或对该资源描述不正确等,可点击“私信”按钮向作者进行反馈;如作者无回复可进行平台仲裁,我们会在第一时间进行处理!
加入交流群
请使用微信扫一扫!