ubuntu怎么装pytorch

在 Ubuntu 上安装 PyTorch，通常按以下步骤进行。下面以**最常见场景（有 NVIDIA 显卡 + CUDA）和无显卡（CPU）**两种情况说明。

一、先确认环境

1. 查看 Ubuntu 版本

lsb_release -a

2. 查看是否有 NVIDIA 显卡

nvidia-smi

• 有输出 → 有 GPU，可用 CUDA
• 命令不存在 → 只能用 CPU 版

二、安装 Python 和 pip

Ubuntu 一般自带 Python3：

python3 --version
pip3 --version

如果没有 pip：

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

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

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

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

四、安装 PyTorch

✅ 有 NVIDIA 显卡（CUDA）

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

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

⚠️ 注意：

• 先装好 NVIDIA 驱动
• CUDA 版本要和驱动匹配

✅ 只有 CPU

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

五、验证安装

python

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

六、常见问题

1. pip 太慢

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

2. 想用 conda

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

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

• Ubuntu 版本
• 是否有显卡
• 是否用 conda

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