首页主机资讯CentOS上PyTorch安装包从哪里下载

CentOS上PyTorch安装包从哪里下载

时间2025-12-12 16:45:03发布访客分类主机资讯浏览1038
导读:下载渠道与推荐路径 官方渠道:使用 PyTorch 的 pip 索引 URL https://download.pytorch.org/whl/,按 CUDA 版本选择子目录(如 cu118、cu121),下载对应的 Linux x86_...

下载渠道与推荐路径

  • 官方渠道:使用 PyTorch 的 pip 索引 URL https://download.pytorch.org/whl/,按 CUDA 版本选择子目录(如 cu118、cu121),下载对应的 Linux x86_64 wheel 包(.whl) 后用 pip 安装。示例目录:https://download.pytorch.org/whl/cu118。该方式适用于 CentOS 7/8/Stream 等常见发行版。
  • 官网页面:访问 https://pytorch.org/get-started/locally/,选择系统、包管理器、Python 与 CUDA 版本,页面会给出可直接复制的 pip/conda 安装命令;如需离线,可复制命令中的下载链接进行手动下载。
  • 历史版本:如需旧版,可在 PyTorch 官网选择 Previous Versions,或直接在 https://download.pytorch.org/whl/ 下选择对应 cuXX 目录查找历史轮子。

离线下载与安装步骤

  • 确认环境信息:在终端执行 nvidia-smi 查看 CUDA Version(如 11.8、12.1),并确保驱动与 CUDA 版本匹配。
  • 选择并下载 wheel:进入对应目录(如 cu121),按 Python 主次版本选择文件,例如 torch-2.2.1%2Bcu121-cp310-cp310m-linux_x86_64.whl(URL 中的 %2B 即加号 +)。也可在官网生成命令后复制链接离线下载。
  • 安装与验证:在目标环境执行 pip 安装并验证
    • pip install torch-2.2.1%2Bcu121-cp310-cp310m-linux_x86_64.whl
    • python - < < ‘PY’ import torch print(torch.version, torch.cuda.is_available()) PY 若输出 PyTorch 版本号且 torch.cuda.is_available() 为 True,则安装成功。

版本匹配与目录选择要点

  • 驱动、CUDA Toolkit 与 PyTorch 版本需匹配;先用 nvidia-smi 确认 CUDA 运行时版本,再选择对应的 cu118/cu121 等目录。
  • wheel 文件名包含关键三元组:例如 cu121(CUDA 版本)、cp310(Python 3.10)、linux_x86_64(平台);必须与环境一致才能安装成功。
  • 无 GPU 或仅需 CPU 版本时,选择 CPU-only 安装方式(pip 安装命令含 cpuonly 或在索引中选择 cpu 目录)。

常见下载目录示例

类别 下载地址 说明
PyTorch Linux wheels(CUDA 12.1) https://download.pytorch.org/whl/cu121 适用于 Python 3.xCUDA 12.1 的 Linux 轮子
PyTorch Linux wheels(CUDA 11.8) https://download.pytorch.org/whl/cu118 适用于 Python 3.xCUDA 11.8 的 Linux 轮子
官网获取安装命令 https://pytorch.org/get-started/locally/ 交互式选择后生成适配命令,便于离线取链接

声明:本文内容由网友自发贡献,本站不承担相应法律责任。对本内容有异议或投诉,请联系2913721942#qq.com核实处理,我们将尽快回复您,谢谢合作!


若转载请注明出处: CentOS上PyTorch安装包从哪里下载
本文地址: https://pptw.com/jishu/770654.html
在CentOS上如何利用PyTorch进行图像识别 Zookeeper如何配置权限控制

游客 回复需填写必要信息