如何用linux安装pytorch

下面给你一份在 Linux 上安装 PyTorch 的常用方法，适用于 Ubuntu / CentOS / Debian 等常见发行版。

一、准备环境（推荐）

1. 安装 Python 和 pip

大多数 Linux 已自带 Python3，确认一下：

python3 --version
pip3 --version

如果没有：

# Ubuntu / Debian
sudo apt update
sudo apt install python3 python3-pip python3-venv

# CentOS
sudo yum install python3 python3-pip

2. 建议创建虚拟环境（可选但推荐）

python3 -m venv pytorch_env
source pytorch_env/bin/activate

二、安装 PyTorch（官方推荐方式）

✅ 方法一：使用 pip 安装（最常用）

1. 查看官方命令

访问：https://pytorch.org/get-started/locally/

2. 常见安装命令示例

CPU 版本

pip3 install torch torchvision torchaudio

CUDA 11.8（有 NVIDIA 显卡）

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

CUDA 12.1

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

✅ 方法二：使用 conda 安装（推荐给科研用户）

conda create -n pytorch python=3.10
conda activate pytorch
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

三、验证是否安装成功

python3

import torch
print(torch.__version__)
print(torch.cuda.is_available())  # True 表示 GPU 可用

四、常见问题

❓ 1. 安装很慢

使用国内镜像：

pip3 install torch torchvision torchaudio -i https://pypi.tuna.tsinghua.edu.cn/simple

❓ 2. CUDA 不可用

• 确认已安装 NVIDIA 驱动
• nvidia-smi 能正常显示
• PyTorch CUDA 版本与驱动匹配

❓ 3. 权限问题

加 --user 或使用虚拟环境：

pip3 install --user torch

五、总结（最省事方案）

python3 -m venv env
source env/bin/activate
pip install torch torchvision torchaudio

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

• 你的 Linux 发行版
• 是否有 NVIDIA 显卡
• 是否用 conda

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