如何在PyTorch的安装教程中加入国内镜像以加速GPU相关的包下载?
时间: 2024-12-12 14:24:47 浏览: 10
在PyTorch的安装过程中加入国内镜像可以显著提高GPU相关的包下载速度。通常,我们推荐使用`torchenv`工具或者修改pip源来实现这个目的。以下是两个方法:
1. **使用`torchenv`**:
- 安装`torchenv`: `pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/cpu/torch_stable.html`
- 激活环境并设置国内镜像:`torchenv create -p http://mirrors.tuna.tsinghua.edu.cn/pypi/simple/`
2. **修改pip源** (适用于系统级别的安装):
- 打开终端,运行:
```
pip config set global.index-url http://mirrors.aliyun.com/pypi/simple/
pip config set global.trusted-host mirrors.aliyun.com
```
或者临时更改:
```
pip install --index-url http://mirrors.aliyun.com/pypi/simple/ pytorch torchvision torchaudio
```
- 如果是Conda环境下,可以在创建新环境时指定镜像源:
```bash
conda create -n myenv python=3.7 -c anaconda -c nvidia
conda activate myenv
conda config --set channels priority true
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
conda install pytorch torchvision cudatoolkit=10.2 -c NVIDIA
```
记得定期检查镜像地址是否需要更新,因为某些软件库可能会有新的版本发布。
阅读全文