pytorch安装

1.先更新nvidia驱动  NVIDIA 引领人工智能计算 | NVIDIA

2.确定要用的cuda型号，下载并安装 （我用的10.2） CUDA Toolkit Archive | NVIDIA Developer

3.去 pytorch官网，获取对应cuda型号的pytorch版本的conda或pip口令  Start Locally | PyTorch

（我用的conda，在conda环境下新建一个pytorch环境，然后运行conda口令）

4.若要更改torchvision版本，在清华源（Index of /anaconda/cloud/pytorch/win-64/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror）下载对应的版本至本地，然后用：conda install --offline 对应的安装包文件名字来安装（如我的是conda install --offline F:\torch+torchvision\torchvision-0.9.1-py36_cu102.tar.bz2）

5.测试：pip list看版本 ；torch.cuda.is_available()检测cuda是否可用