PyTorch安装教程

PyTorch安装教程

1. 在Pytorch官网中选择想要安装的pytorch版本（一般是根据项目需求来确定）。

2. 选定Pytorch版本后，再选定CUDA版本，如下CUDA 11.8，代表CUDA版本为11.8，记住这个CUDA版本号，

3. 在Anaconda中新建虚拟环境。conda create -n XXX python=3.9（python版本和pytorch版本有对用关系，网上可以找到）

4. 在虚拟环境中安装CUDA 和 CUDNN。 先使用命令 conda install cudatoolkit=11.8， 再使用命令conda install cudnn 这个命令不需要指定版本号，会根据cuda版本进行适配

5. 安装pytorch，在官网中，找得对应版本的pip 安装命令。

直接复制命令，安装即可。

pip install torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 --index-url https://download.pytorch.org/whl/cu118

6. 验证

在虚拟环境中进入python环境，
import torch # 导入torch包
torch.cuda.is_available() # 判断torch的cuda是否可用。若为True,则安装成功。