linux pytorch怎样安装依赖

linux

小樊

2026-10-19 02:46:25

栏目: 智能运维

在 Linux 上安装 PyTorch 依赖，通常分为 系统依赖 和 Python 依赖 两部分。下面按最常见场景说明（以 Ubuntu / Debian 为例）。

一、系统基础依赖（编译/运行需要）

sudo apt update
sudo apt install -y \
    python3 \
    python3-pip \
    python3-venv \
    build-essential \
    libgl1 \
    libglib2.0-0

如果你要用 GPU（CUDA），还需安装：

可用官方脚本或：

nvidia-smi

确认驱动是否正常。

python3 -m venv pytorch-env
source pytorch-env/bin/activate

pip install torch torchvision torchaudio

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

具体 CUDA 版本请参考：
https://pytorch.org/get-started/locally/

如果项目有 requirements.txt：

pip install -r requirements.txt

常见依赖示例：

numpy
matplotlib
opencv-python
tqdm
pandas
scikit-learn

安装命令：

pip install numpy matplotlib opencv-python tqdm pandas scikit-learn

import torch
print(torch.__version__)
print(torch.cuda.is_available())

✅ pip 版本过低

pip install --upgrade pip

✅ 权限问题

pip install --user ...

或使用虚拟环境（推荐）

✅ 需要编译扩展（如 detectron2）

sudo apt install -y ninja-build

如果你能告诉我：

我可以给你精确的一条命令安装方案。