配置pytorch

删除虚拟环境：

conda remove -n pytorch --all

创建虚拟环境：

conda create -n pytorch python=3.8

激活虚拟环境：

conda activate pytorch

打开pytorch官网选择版本
https://pytorch.org/
这里我是11.3

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch