安装pytorch

下载地址是我！

指定链接下载指定torch版本

pip install torch==1.1.0--find-links https://download.pytorch.org/whl/torch_stable.html

• cu115代表CUDA版本是11.5
• cp38代表CUDA版本是3.8

• 先点击下载链接，将安装包下载到本地
• 使用pip安装

pip install torch-1.11.0-2Bcu115-cp38-cp38-linux_x86_64.whl

安装torchvision

pip install torchvision==0.2.0

torch 和 torchvision对应版本

来源：https://www.cnblogs.com/lyl0618/p/13323010.html

分别安装指定版本的命令

conda install pytorch==1.1.0 torchvision==0.11.1 cudatoolkit=11.5 -c pytorch