PyTorch Linux安装包在哪里下载
小樊
43
2025-12-14 13:44:21
PyTorch Linux安装包下载渠道
官方下载渠道
- pip 轮子索引目录:https://download.pytorch.org/whl/torch_stable.html(包含 torch / torchvision / torchaudio 的 Linux 轮子,按 CUDA 版本、Python 版本、manylinux 标签组织,适合离线下载后用 pip 安装)。
- 官方安装指引页:https://pytorch.org/get-started/locally(选择 Linux、包管理器 pip/conda、以及 CPU/CUDA 版本,页面会给出对应命令;如需历史版本,可点页面中的 Previous PyTorch Versions 链接进入 https://pytorch.org/get-started/previous-versions/)。
离线下载与安装步骤
- 在上面的索引页或指引页确定并下载匹配你环境的文件,例如:torch-1.13.0+cu116-cp37-cp37m-linux_x86_64.whl(含义:PyTorch 版本、CUDA 版本、Python 版本、系统架构)。
- 将 .whl 上传到 Linux 服务器后执行:pip install torch-1.13.0+cu116-cp37-cp37m-linux_x86_64.whl。
- 如需同时离线安装 torchvision/torchaudio,务必选择与之完全匹配的版本组合(同一 PyTorch 版本 + CUDA 标签),避免运行时依赖冲突。
conda 渠道
- 使用 Anaconda/Miniconda 可直接从官方通道安装:conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch(将 cudatoolkit 版本替换为你的 CUDA 版本)。
- 如需离线 conda 包,可在 Anaconda 镜像查找对应 channel(如 pytorch 的 conda 仓库),但镜像可能存在同步延迟或失效,优先使用官方通道更稳妥。
镜像与国内加速
- 清华 Conda 镜像:https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/(适合 conda 离线缓存;若遇到失效,可回退到官方源)。
- PyTorch 官方 whl 索引:https://download.pytorch.org/whl/torch_stable.html(亦可结合 wget/curl 直接下载所需轮子)。
选择与校验要点
- 先确认 Python 版本(如 cp38/cp39/cp310)、系统架构(如 x86_64 或 aarch64)、以及 CUDA 版本(如 cu116/cu117/cu121 或 cpu)。
- 轮子文件名通常包含这些关键信息,例如:torch-2.1.0-cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.whl;安装时保持 torch / torchvision / torchaudio 三者的版本与 CUDA 标签一致。
- 在 ARM64(aarch64) 服务器上,官方亦提供 manylinux 轮子,可直接用 pip 安装(示例见昇腾社区给出的 manylinux2014_aarch64 / manylinux_2_17_aarch64 包)。