windows安装pytorch前注意事项

（是根据多个帖子总结的，可能有误）

（默认电脑已安装anaconda，最终配置是win11+cuda12.1.0+cudnn8.9.3+python3.10+pytorch2.3.0）

1.pytorch分为cpu版本和gpu版本

在设备管理器的显示那，如果有NVIDIA的显卡，则适合安装gpu版本；没有则是cpu版。

下面讲的是gpu版的安装

2.gpu版还需要安装cuda和cudnn（前者是一个方便pytorch计算的工具包，后者是前者的扩展库）

2.1 cuda的安装

首先要确定自己要下载什么版本，通过控制面板得到一个驱动的版本：

安装的cuda版本不能超过该版本

官网地址：https://developer.nvidia.com/cuda-toolkit-archive

在安装前，若是开发c++的，得先安装好visual studio（注意版本不要与cuda版本冲突），若是用pycharm的则不要求。

本帖安装的12.1（不要装12.0，有坑）

临时地址不用改，选自定义

下两个组件若新版本小于等于当前版本，则不要勾选。后面就点点点

安装结束后，通过控制面板验证（要重启）

2.2 cudnn的安装

下载路径（需要登陆）：cuDNN Archive | NVIDIA Developer

选与cuda对应版本的，下载完得到一个压缩包，将解压后的三个文件夹复制到cuda中（我是分别复制的）

而后添加环境变量（手动增加最后两行）

在控制面板运行两exe文件验证，结果都是PASS证明成功。

3.pytorch-gpu安装

官网：https://pytorch.org/

（清华源只有cpu版本）

若找不到对应版本，也可以选择下torch+torchaudio+torchvision三者的whl文件

download.pytorch.org/whl/torch_stable.html

分别代表cuda版本，torch（另外两个往下翻）版本，python版本，操作系统。（各版本注意要互相对应，下图示例）

新建一个虚拟环境进行安装（看个人选择）

conda create -n pytorch3.10 python=3.10
conda activate pytorch3.10
conda install pytorch==2.3.0 torchvision==0.18.0 torchaudio==2.3.0 pytorch-cuda=12.1 -c pytorch -c nvidia

安装后验证，true则成功。

python
import torch
torch.__version__
torch.cuda.is_available()