当前位置: 首页 > 图灵资讯 > 行业资讯> 如何使用pip安装pytorch

如何使用pip安装pytorch

来源:图灵python
时间: 2026-09-03 16:17:51
pip install torch 主要原因是Python版本不兼容(如3.12+或3.7-)、系统架构/平台不匹配(如M1误用X86命令)、pip过旧或未指定Pytorch官方索引源。

pip install torch 为什么命令经常失败?

直接运行 pip install torch 大概率会安装 CPU 即使你有版本 NVIDIA 因为显卡 pip 默认不检测 CUDA 环境不会自动匹配你当地的驱动和 CUDA Toolkit 版本。更常见的是报错:ERROR: No matching distribution found for torch,这通常是因为你 Python 版本太新(例如 3.12+)或太旧(

  • PyTorch 2.4+ 要求 Python ≥3.8,推荐用 3.10 或 3.11
  • Windows 用户若用 PowerShell,记得先运行 Set-ExecutionPolicy RemoteSigned -Scope CurrentUser,否则,系统策略可能会阻止虚拟环境的激活
  • macOS M 不能使用系列芯片用户 cu118 等 CUDA 链接——M 芯片没有 CUDA,只能装 torch 的原生 Apple Silicon 版本(含 MPS 支持)
怎么选对 --index-url 参数

关键不是“有没有” GPU而是“你的 NVIDIA 哪一个可以支持驱动 CUDA 版本”。运行 nvidia-smi 检查右上角显示的内容 “CUDA Version: 12.2” ——这是驱动**最高支持**的最大支持 CUDA 版本,不是你安装的版本 CUDA Toolkit 版本。你必须去 CUDA Toolkit 对照表的官方版本 检查你实际装的 Toolkit 版本(如 nvcc --version),再回 PyTorch 官网 生成安装命令。

  • 驱动支持 CUDA 12.2,但你只装了 CUDA 11.8 Toolkit → 用 --index-url https://www.tulingpython.com/d/file/p/20260903/mpq3xv0ubar
  • 只支持驱动 CUDA 11.2 → 即使官网有 cu118 链接,也不能安装,会提示 libcudart.so.11.8: cannot open shared object file
  • 用 conda 装过旧版 PyTorch?先运行 pip uninstall torch torchvision torchaudio,conda 和 pip 很容易造成混装 ImportError: libcudnn.so.8: cannot open shared object file
验证 torch.cuda.is_available() 返回 False 怎么办

这不是安装失败,而是环境在运行过程中没有连接 GPU。常见原因及检查点:

  • 验证脚本未在激活的虚拟环境中运行(例如,您在激活的虚拟环境中 base 环境里装了 torch,却在另一个 venv 里 import)
  • Linux 下忘记加 export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH~/.bashrc
  • Windows 上 CUDA Toolkit 系统未加入安装路径 PATH(应包含 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\bin
  • 代码里漏了 torch.device("cuda"),误以为 is_available() 为 True 就自动用 GPU ——它只是告诉你“可用”,并不意味着张量默认为 GPU 上

验证时必须使用这个最小代码:

CSV Data Pipeline

CSV和JSON数据的处理、转换和分析,支持CSV的筛选、合并、聚合、格式转换、去重和摘要报告、TSV和JSON Lines文件。

下载

import torch
print(torch.__version__)
print(torch.cuda.is_available())
print(torch.cuda.device_count())
if torch.cuda.is_available():
    print(torch.cuda.get_device_name(0))
国内用户无法回避的镜像和权限问题

清华源,阿里云源对 PyTorch wheel 延迟同步(通常) 1–3 天),所以不要在那里 PyTorch 新版本发布当天更换源安装。临时使用官方来源更稳定:

  • 临时指定官方来源:pip install torch --index-url https://www.tulingpython.com/d/file/p/20260903/mpq3xv0ubar --extra-index-url https://pypi.org/simple
  • Windows 上遇到 Access is denied:别加 sudo(无用),改用 pip install --user torch 安装在用户目录中
  • Mac 上 pip 报 zsh: command not found: pip:说明你用的是系统自带的 Python(已弃用),请使用 brew install python 安装新版,再用 pip3;或者确认 which python3which pip3 输出路径一致

最容易被忽视的一点:PyTorch 的 torchaudiotorchvision 必须跟 torch 严格对齐版本。手动指定版本时,三包要写全,例如:pip install torch==2.4.0 torchvision==0.19.0 torchaudio==2.4.0 --index-url https://www.tulingpython.com/d/file/p/20260903/mpq3xv0ubar。少一个,后续 import 就可能出 Segmentation fault。