在CUDA环境下安装Pytorch【简洁版】

1、查看自己的CUDA版本

在命令提示符中输入：

nvidia-smi.exe 

下图为例，CUDA Veision: 12.2

2、选择与CUDA版本适配的Pytorch版本号

在浏览器中输入：

https://pytorch.org/get-started/previous-versions/

下图为例，因为CUDA Veision: 12.2，因此选择 # CUDA 12.1下的指令

直接在命令提示符中输入：

conda install pytorch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 pytorch-cuda=12.1 -c pytorch -c nvidia 

结果如下：

然后静静等待……