一、下载相关whl
1、从Previous PyTorch Versions | PyTorch检查相互版本是否兼容,否则会出现无法使用cuda的问题;
2、从https://download.pytorch.org/whl/torch_stable.html下载torh,torchvision,torchaudio等
cu117/torch-2.0.0%2Bcu117-cp38-cp38-win_amd64.whl
cu117/torchaudio-2.0.0%2Bcu117-cp38-cp38-win_amd64.whl
cu117/torchvision-0.15.0%2Bcu117-cp38-cp38-win_amd64.whl
3、下载到任意目录,进入此任意目录执行如下命令
pip install "torch-2.0.0+cu117-cp38-cp38-win_amd64.whl" -i http://pypi.douban.com/simple --trusted-host pypi.douban.com
pip install "torchaudio-2.0.0+cu117-cp38-cp38-win_amd64.whl" -i http://pypi.douban.com/simple --trusted-host pypi.douban.com
pip install "torchvision-0.15.2+cu117-cp38-cp38-win_amd64.whl" -i http://pypi.douban.com/simple --trusted-host pypi.douban.com
4、检查是否安装成功
import torch
torch.__version__
torch.cuda.is_available()