pytorch、cudnn、cuda安装教程

pytorch、cudnn、cuda安装教程

一、cuda安装

1. cmd输入命令nvidia-smi查看自己电脑支持的CUDA版本
   我的显示如下：我的电脑上的GPU最高支持的CUDA版本是12.4：
   
2. 打开网址：https://developer.nvidia.com/cuda-toolkit-archive
   根据自己的刚才查看到的CUDA版本,选择一个适合自己的版本：
   
   点开后: 根据自己的电脑选择平台，我是win11，就选11，安装类型选择local，不要选择network,network表示通过网络安装
   
3. 双击下载下来的exe文件，
   ①安装路径不要更改、就是用默认的
   
   ②后面一路同意，选择安装模式的时候要选择自定义模式
   
   ③在自定义安装选项中，点开CUDA选项，取消Visual Studio Integration选项，再一直点下一步就可以了
   

二、安装cudnn

cudnn是用来激活cuda的，必须安装

1. 进入C盘我们刚才安装CUDA的安装目录：
2. 进去会看到我们刚才安装好的CUDA 11.8
   
3. 下载cudnn：
   ① 打开链接：https://developer.nvidia.cn/cudnn
   ② 点击下载，不要点上面的下载，点下面的
   
   ③ 接下来会要求你注册，注册登录即可
   ④ 看你的CUDA是什么版本，我的是11.8，就点击for CUDA 11.x
   
   ⑤ 选择windows版本进行下载
   
   ⑥ 下载下来后是一个压缩包、解压
   将里面的前三个目录复制到我们安装CUDA的位置中去、选择全覆盖
   
   最后CUDA安装目录中就会包含CUDNN的内容
   
4. 设置环境变量
   将CUDA安装路径下的bin、lib、libnvvp、include全部添加到path中，至此CUDA安装完成

安装pytorch

1. 进入https://pytorch.org/get-started/previous-versions/网址，选择合适自己的pytorch版本和安装方式、复制安装命令执行即可
   我是使用conda安装，且需要2.0.1版本的pytorch，所以我是用的下面的命令