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