1.Conda创建虚拟环境
conda create -n pytorch_gpu python=3.6
2.Pip安装pytorch
pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
3.测试pytorch安装
python
import torch
torch.__version__