快速安装Pytorch

一、查看CUDA版本

Linux:

nvcc --version

Windows:

nvcc --version

二、在官网找到合适的Pytorch版本

官网：PyTorch

1.进入官网后直接下滑

2.按照自己的要求，选择合适的版本，复制Command

3.若官网下载速度过慢可添加后缀来更换下载源，如下所示：

pip3 install torch torchvision torchaudio -i https://mirrors.aliyun.com/pypi/simple/