在国内下载PyTorch时,可以通过以下方法提高下载速度:
1. 使用国内镜像源
国内镜像源能显著加快下载速度,常用的镜像源有:
-
清华源
pip install torch==2.3.1 torchaudio==2.3.1 torchvision==0.18.1 -f https://mirrors.tuna.tsinghua.edu.cn/pytorch-wheels/torch_stable.html
-
阿里云源
pip install torch==2.3.1 torchaudio==2.3.1 torchvision==0.18.1 -f https://mirrors.aliyun.com/pytorch-wheels/torch_stable.html
2. 使用pip
的-i
选项
直接指定镜像源:
pip install torch==2.3.1 torchaudio==2.3.1 torchvision==0.18.1 -i https://pypi.tuna.tsinghua.edu.cn/simple
3. 使用代理
如果有代理服务器,可以通过以下命令设置代理:
pip install torch==2.3.1 torchaudio==2.3.1 torchvision==0.18.1 --proxy=http://your_proxy:your_port
4. 提前下载whl文件
手动下载whl文件并使用pip
安装:
- 访问PyTorch官网下载对应版本的whl文件。
- 使用
pip
安装下载的文件:
pip install /path/to/torch-2.3.1-cpXX-cpXX-XXXXXX.whl
5. 使用conda
(如果已安装)
conda
通常比pip
更快:
conda install pytorch==2.3.1 torchaudio==2.3.1 torchvision==0.18.1 -c pytorch
通过这些方法,可以显著提升下载速度。