anaconda安装cuda cudnn pytorch（win11）

anaconda安装cuda cudnn pytorch（win11）

在anaconda环境中安装的cuda，如果直接在电脑里安装cuda，本文仅提供参考意义

一、确定对应版本

1、确定cuda版本

1、win+r 打开运行窗口
2、输入cmd打开终端
3、输入 nvidia-smi 查看显卡驱动版本（Driver Version）和其支持的cuda版本（CUDA Version），这里的cuda版本指的是该显卡驱动支持的最新cuda版本，cuda是可以兼容旧版本的。比如我是 12.5，那么比12.5低的都可以安装。

2、确定torch版本

pytorch网址
下拉查询

这里可以看到对python的版本也是有要求的，新的pytorch要求python版本最低是3.8。python比较好下载，这里可以查询别的教程。需要别的版本的pytorch可以点击左下。

这个网页暂时不要关闭，一会我们需要复制命令。

二、安装cuda、pytorch和cudnn

1、安装cuda

cuda下载网址

点击该网址，找到对应的cuda版本

激活想要安装cuda和pytorch的对应conda环境，输入以下代码安装cuda，注意修改你对应的cuda版本号

conda install cuda -c nvidia/label/cuda-12.4.0

这里如果下载速度特别慢，你需要科学上网。

如果在科学上网之后还是卡在了某个包上，终止下载，输入以下命令清除下载缓存以后重新下载。

conda clean --all

下载完成应该出现以下文字：

2、验证cuda下载正确可用：

方法一：
激活对应conda环境，输入

nvcc -V

有如下显示即可：

方法二：

激活对应conda环境

运行cuda自带的demo文件进行测试。新版的cuda自带的这两个文件的目录参考这里 D:\anaconda3\envs\pytorch2_4_0\demo_suite

下面是进入目录和运行测试的代码：（windows终端操作）

d: //切换磁盘
cd D:\anaconda3\envs\pytorch2_4_0\demo_suite //进入目标目录
bandwidthTest.exe //测试程序1
deviceQuery.exe //测试程序2

两个测试都显示pass即可

3、安装pytorch

激活对应conda环境

输入刚才查询pytorch版本时，下面那段代码

conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia

显示 done就是完成了

4、安装cudnn

1、查询cudnn和cuda的版本对应关系：
https://developer.nvidia.com/rdp/cudnn-archive

2、点开以后选择对应windows的.zip压缩包

这里需要你注册一个NVIDIA账号，自行注册即可

下载完成后，解压完应该是这样：

3、将解压完的文件目录下的文件复制到你的conda环境下的对应文件夹中：
我的路径是 D:\anaconda3\envs\pytorch2_4_0 （这里的pytorch2_4_0是我的环境名）

这里lib文件夹下是还有x64文件夹的，找到对应文件夹即可。

4、添加环境变量

在系统变量的path中添加以下路径：

D:\anaconda3\envs\pytorch2_4_0\bin

5、验证cudnn安装成功

这个具体我没找到方法，官方的cudnn安装文档也没有提供验证方法。

这里希望有人找到明确的方法，望告知！！！

我跑了以下代码，成功了就没管了

from torch.backends import cudnn
import torch
print(cudnn.is_available())  #返回True说明已经安装了cuDNN

print(torch.__version__)
print(torch.cuda.is_available())

ps：实际上我删除了复制的cudnn文件，这里还是可以跑通。

三、结束

希望对大家配置环境有所帮助，共勉。

附件：

NVIDIA的Windows CUDA官方安装指南
官方的cudnn安装文档