一、pytorch环境搭建
从官网pip安装速度太慢，这里介绍一种快速安装方法，首先打开在终端中输入
nvidia-smi
查看cuda版本，注意11.4的可以装11.1 11.2 不一定得是11.4；
其次去pytorch官网确定要安装的pytorch ，例如安装 ：
pip install torch==1.10.0+cu111 torchvision==0.11.0+cu111 torchaudio==0.10.0 -f https://download.pytorch.org/whl/torch_stable.html
接着去pytorch whl源 ，下载要安装的torch、torchvision、torchaudio，注意版本要对应；
下载完成直接 pip install xxx/.whl；
最后检测pytorch安装成功
import torch torch.cuda.is_available() 返回True则表示安装成功
