最近配torch 环境 官网命令安装老是断线,所以采用了离线安装方式,顺便写个博客,大家遇到同一问题可参考:

网址: https://download.pytorch.org/whl/torch_stable.html

1. 下载合适版本(比如我本机是cuda11.6)

torch-1.12.0+cu116-cp38-cp38-win_amd64.whl

torchaudio-0.12.0+cu116-cp38-cp38-win_amd64.whl

(注意下面的torchvision 要比torch 高一个版本,比如torch 是1.12, torchvison 要是0.13)

torchvision-0.13.0+cu116-cp38-cp38-win_amd64.whl   

如:

2.pip 安装(导航到三个轮子的根目录,然后cmd 激活虚拟环境执行以下命令)

pip install torch-1.12.0+cu116-cp38-cp38-win_amd64.whl
pip install torchaudio-0.12.0+cu116-cp38-cp38-win_amd64.whl

pip install path\to\torchvision-0.13.0+cu116-cp38-cp38-win_amd64.whl
 

3. 验证:

python -c "import torch; print(torch.__version__)"
python -c "import torchaudio; print(torchaudio.__version__)"
python -c "import torchvision; print(torchvision.__version__)"

成功

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐