安装torch清华镜像
时间: 2024-12-31 18:44:02 浏览: 10
### 如何通过清华大学镜像源安装 PyTorch
#### 配置清华镜像源
为了加速 PyTorch 的安装过程并提高下载速度,可以使用清华大学开源软件镜像站作为 pip 或者 conda 的源。
对于 `pip` 用户,在命令行中加入 `-i https://pypi.tuna.tsinghua.edu.cn/simple` 参数来指定使用清华镜像源[^2]:
```bash
pip install torch torchvision torchaudio -i https://pypi.tuna.tsinghua.edu.cn/simple
```
而对于 Conda 用户,则可以通过编辑 `.condarc` 文件或直接运行如下命令添加清华镜像源中的 PyTorch 渠道[^3]:
```bash
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
```
#### Linux 上的一键 Shell 脚本安装方法
针对 Linux 系统用户,还可以编写简单的 shell 脚本来实现自动化安装流程。下面是一个示例脚本[^1]:
```bash
#!/bin/bash
# 安装 PyTorch 及其依赖项
pip3 install torch torchvision torchaudio -i https://pypi.tuna.tsinghua.edu.cn/simple
```
保存上述内容到文件(如 `install_pytorch.sh`),赋予执行权限并通过终端运行该脚本即可完成安装。
测试安装结果以确认一切正常工作是非常重要的一步。通常情况下,可以在 Python 解释器内尝试导入刚刚安装好的库来进行验证。
阅读全文