安装PyTorch的CUDA版本时，需要确保从官方仓库检查是否有对应版本的软件包。例如，对于CUDA10.2，应安装torch1.10.2+cu102、torchvision0.11.3+cu102和torchaudio0.10.2+cu102。如果遇到依赖不兼容问题，可能需要调整版本号以达到匹配。
PyTorch安装CUDA版本的正确方式
很多人按照PyTorch官方的说明安装CUDA版本，装完却发现装的是CPU版本。
原因就是是仓库里就没有这个版本的软件包，安装前需要在下面的网址查看一下儿是否有编译好的软件包：
- https://download.pytorch.org/whl/torch_stable.html
- https://download.pytorch.org/whl/cu102/torch_stable.html
以安装PyTorch CUDA10.2版本为例，先安装torch，发现仓库里只有torch的1.10.2版本。
再安装torchvision和torchaudio，如后两个有依赖不兼容，可以调整小版本号。
示例：
pip install torch==1.10.2+cu102 torchvision==0.11.3+cu102 torchaudio==0.10.2+cu102 -f https://download.pytorch.org/whl/torch_stable.html
扫一扫
9621
到【灌水乐园】发言
