个人总结pytorch安装不易踩坑版本

1.直接在命令提示行输入nvidia-smi查询cuda版本。

2.python版本建议：3.8.10 （64bit）；为什么不说别的，是因为之前用过python3.8.0，3.11，3.10版本都报过不一样的错。

 python及自用torch版本

3.在pytorch官网找符合自己版本的pip命令，直接pip

Previous PyTorch Versions | PyTorch

4.若是pip报错或者下成cpu版本，可以复制pip后面跟着的网站，但都可以ctrl+f 搜索cu +自己的cuda版本，比如cuda 10.2 就是cu102 ，找相应的torch,torchvision,torchaudio版本，比如上图中的CUDA 10.2 对应的版本分别是 torch 1.12.1+cu102 , torchvision 0.13.1+cu102 , torchaudio 0.12.1

很明显，这个版本的torch只有linux版本，往下翻找到有win系统的torch版本，对应下载，cp38就是对应python 3.8 版本。

5.进入存有上述文件的目录，输入cmd并回车，进入命令提示符

6.用pip install  xxx.whl 安装，安装torchvision的时候可能需要pillow, numpy 等依赖库，下载可能会超时。

pip install pillow -i https://pypi.tuna.tsinghua.edu.cn/simple