打开pytorch官网
选择对应选项，我的cuda是11.2版本的，选择相近的11.1

把底下的命令copy去cmd执行即可安装
conda：

conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge

pip：

pip3 install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio===0.9.0 -f https://download.pytorch.org/whl/torch_stable.html