安装pytorch

直接通过pip安装pytorch，cuda版本为11.3

pip3 install torch==1.10.0+cu113 torchvision==0.11.1+cu113 
torchaudio==0.10.0+cu113 -f 
https://download.pytorch.org/whl/cu113/torch_stable.html

其他安装方法参照官网

https://pytorch.org/get-started/locally/