pytorch安装(解决网络慢)
一、确定pytorch版本
# CUDA 11.6
pip install torch==1.12.0+cu116 torchvision==0.13.0+cu116 torchaudio==0.12.0 --extra-index-url https://download.pytorch.org/whl/cu116
二、解决网速过慢
1、下载(链接)到本地
- 需要下载:torch、torchaudio、torchvision
- 下载的版本如下所示
- cu116:cuda 11.6
- cp37:python=3.7
2、安装
pip install ./"torch-1.12.0+cu116-cp37-cp37m-win_amd64.whl"pip install ./"torchaudio-0.12.0+cu116-cp37-cp37m-win_amd64.whl"pip install ./"torchvision-0.13.0+cu116-cp37-cp37m-win_amd64.whl"
参考:[PyTorch] 安装笔记, 基于Windows10/cuda11.6 - 知乎 (zhihu.com)
到此,pytorch安装完毕!!!