安装pytorch和torch

conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=11.3 -c pytorch -c conda-forge

pip install torch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1

##注意torch一定要和torchvision搭配使用，版本对应。否则会出现No moudle 'torch'错误。可以同conda list 查看自己的库包。发现存在torch。