pytorch 离线安装
(注意下面的torchvision 要比torch 高一个版本,比如torch 是1.12, torchvison 要是0.13)2.pip 安装(导航到三个轮子的根目录,然后cmd 激活虚拟环境执行以下命令)1. 下载合适版本(比如我本机是cuda11.6)
最近配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__)"
成功
更多推荐
所有评论(0)