小白安装CUDA-cuDNN

(仅供自己学习总结用)

背景:电脑重置后要自己重新装一次pytorch框架。

附加:NVIDIA(显卡)和CUDA(驱动)的版本匹配也是有讲究的

官方文档:https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html(官方文档看得辛苦,也可以看大佬们整理出来的结论啊,哈哈)

参考博客:https://www.jianshu.com/p/978bc51029fa

参考博客:https://blog.csdn.net/qq_58611650/article/details/123450460

 

操作系统,以及对应Driver Version的值

 

关于pytorch的选择,

一定要看好Pytorch最高能支持的CUDA的版本。

首先查看一下自己的电脑是否装了CUDA

参考博客:https://blog.csdn.net/m0_64346597/article/details/125822005

看箭头的位置,有就是有,没有就要下载。

如果有,就直接去下载cuDNN,下载适合自己CUDA的cuDNN.

一、查看电脑支持的CUDA版本

参考博客:https://blog.csdn.net/Jin1Yang/article/details/124754015

箭头所指那一行的CUDA11.6

二、进入CUDA的官网https://developer.nvidia.com/cuda-toolkit-archive,最上边的是最新的,其他版本自己选择。

参考博客:https://blog.csdn.net/Jin1Yang/article/details/124754015

按需下载,不一定说最新的最好。

 

 

三、安装完成CUDA,就要去找合适的cuDNN

cuDNN下载链接:https://developer.nvidia.com/rdp/cudnn-download

参考博客:https://blog.csdn.net/Jin1Yang/article/details/124754015

第一次下载,需要注册一个账号,如果没有是不能下载cudnn的。注册大概需要10分钟左右。

看情况选择。比如参考博客中,

  1. 根据自己安装的cuda版本选择cudnn,这里我们安装的是11.5,所以下载11.5版本

解压缩下载的cuDNN的压缩包:

我选择精简安装,大概就是默认安装,装在了C盘,我的安装地址为:

 

检查是否安装成功:

进入这个地址,复制,然后去cmd窗口

使用cd命令,进入到demo_suite文件夹,再输入bandwidthTest.exe回车,结束后,同样方法输入deviceQuery.exe回车。

都得到以下:

两个PASS,代表成功安装。

(ps:以上图片非全部原创,大部分直接截取参考博客里的图片)小白通过参考以上的博客和文档成功实现安装。

 

posted @ 2022-11-23 17:29  羽落离殇  阅读(300)  评论(0)    收藏  举报