更新pytorch版本镜像源
时间: 2025-01-04 09:20:26 浏览: 5
### 如何更新 PyTorch 版本并设置镜像源
#### 更新 PyTorch 版本
为了确保使用的是最新的 PyTorch 版本,可以通过 Conda 或 Pip 来执行更新操作。
对于通过 Conda 安装的用户,在终端或 Anaconda Prompt 中运行如下命令来更新 PyTorch 到最新版本:
```bash
conda update pytorch torchvision torchaudio -c pytorch
```
如果希望指定特定 CUDA 版本,则可以添加 `cudatoolkit` 参数。例如,要安装带有 CUDA 11.0 的 PyTorch 可以这样写:
```bash
conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch
```
而对于采用 Pip 进行管理的情况,可利用以下指令完成升级工作:
```bash
pip install --upgrade torch torchvision torchaudio
```
#### 设置镜像源加速下载过程
考虑到网络状况可能影响到包文件获取的速度甚至成功率,推荐配置国内镜像源作为解决方案之一。这里介绍两种常用的方式——Conda 和 Pip。
##### 使用 Conda 添加清华镜像源
在 Anaconda Prompt 下依次键入下列语句以便将清华大学开源软件镜像站设为默认渠道[^5]:
```bash
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes
```
注意:当从清华镜像源下载 PyTorch 时不需要再额外指明 `-c pytorch` ,因为在上述配置里已经包含了该选项[^2]。
##### 使用 Pip 指定清华镜像源
对于偏好于 Pip 工具的朋友来说,可以在每次调用时附加参数指向所需服务器地址;亦或是修改全局配置使得所有后续请求自动应用此设定。具体做法是在命令后面追加 `-i` 参数加上对应 URL 地址即可实现临时变更效果[^4] :
```bash
pip install torch torchvision torchaudio -i https://pypi.tuna.tsinghua.edu.cn/simple
```
阅读全文