如何安装debian pytorch

debian

小樊

2026-10-20 01:00:05

栏目: 智能运维

下面分 “安装 Debian 系统上的 PyTorch” 和 “在 Debian 下推荐的安装方式” 两部分说明，适用于 Debian 11 / 12。

一、先确认环境

1. 系统版本

cat /etc/debian_version

2. Python 版本（推荐 3.9+）

python3 --version

Debian 12 默认是 Python 3.11，Debian 11 是 3.9

二、推荐方式：使用虚拟环境安装（最安全）

1. 安装基础依赖

sudo apt update
sudo apt install -y python3-venv python3-pip

2. 创建虚拟环境

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

三、安装 PyTorch（CPU 版）

适合 无 NVIDIA 显卡 或只是学习使用：

pip install torch torchvision torchaudio

验证：

python -c "import torch; print(torch.__version__)"

四、安装 PyTorch（GPU / CUDA 版）

1. 确认有 NVIDIA 显卡

nvidia-smi

2. 安装 NVIDIA 驱动（Debian）

sudo apt install -y nvidia-driver

重启后确认：

nvidia-smi

3. 安装对应 CUDA 的 PyTorch（示例 CUDA 12.1）

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

其他版本见：https://pytorch.org/get-started/locally/

验证：

python -c "import torch; print(torch.cuda.is_available())"

输出 True 即成功。

五、常见问题

1. pip 版本太旧

pip install --upgrade pip

2. 国内加速（可选）

pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

3. Debian 系统 Python 不建议直接装包

✅ 使用 venv
❌ 避免 sudo pip install

六、最小验证示例

import torch

x = torch.rand(3, 3)
print(x)
print("CUDA:", torch.cuda.is_available())

如果你告诉我：

Debian 版本
是否有 GPU
用于学习 / 训练 / 部署

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