怎么在linux用pytorch

linux

小樊

2026-10-04 11:32:43

栏目: 智能运维

在 Linux 上使用 PyTorch，一般流程包括：安装 Linux 环境 → 安装 Python → 安装 PyTorch → 验证 → 写代码运行。下面给你一个清晰、实用的指南。

一、准备 Linux 环境

常见发行版：

Ubuntu / Debian（最推荐，资料最多）
CentOS / Rocky / Alma
Arch

如果是服务器，一般已经是 Linux，无需重装。

查看系统：

uname -a
cat /etc/os-release

二、安装 Python（推荐 3.9–3.11）

Ubuntu / Debian

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

查看版本：

python3 --version

三、创建虚拟环境（强烈推荐）

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

激活后命令行前面会出现 (pytorch-env)

四、安装 PyTorch（关键）

1️⃣ 有 NVIDIA GPU（CUDA）

先确认显卡驱动：

nvidia-smi

去官网选命令（示例 CUDA 12.1）：

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

2️⃣ 只有 CPU

pip install torch torchvision torchaudio

五、验证是否安装成功

python

import torch
print(torch.__version__)
print(torch.cuda.is_available())  # GPU 为 True

六、写一个简单的 PyTorch 程序

test.py

import torch

x = torch.rand(3, 3)
y = torch.rand(3, 3)
print(x + y)

if torch.cuda.is_available():
    x = x.cuda()
    print("Using GPU")

运行：

python test.py

七、常见问题

❌ pip 太慢

pip install torch -i https://pypi.tuna.tsinghua.edu.cn/simple

❌ CUDA 不可用

驱动没装
PyTorch CUDA 版本不匹配
用 nvidia-smi 和 torch.version.cuda 对比

八、进阶建议

Jupyter：pip install jupyterlab
深度学习框架：Transformers、Lightning
多卡训练：DistributedDataParallel

如果你愿意，可以告诉我：

是 服务器还是笔记本
有没有 NVIDIA 显卡
用的是 Ubuntu 还是 CentOS

我可以直接给你一条最适合你的安装命令。