ubuntu pytorch如何安装

在 Ubuntu 上安装 PyTorch 通常推荐用 conda 或 pip，并且要根据你是否有 NVIDIA GPU 来选择 CPU 版或 CUDA 版。

一、确认系统环境

1. 查看 Ubuntu 版本

lsb_release -a

2. 查看是否有 NVIDIA GPU

nvidia-smi

• 有输出：支持 CUDA
• 无输出：只能用 CPU 版

3. 查看 Python 版本

python3 --version

建议使用 Python 3.8–3.11

二、使用 pip 安装（推荐新手）

✅ CPU 版

pip install torch torchvision torchaudio

✅ GPU 版（CUDA 11.8 示例）

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

CUDA 12.1 可替换为 cu121

三、使用 conda 安装（推荐科研/深度学习）

1. 安装 Miniconda

wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
bash Miniconda3-latest-Linux-x86_64.sh

2. 创建环境

conda create -n pytorch python=3.10
conda activate pytorch

3. 安装 PyTorch

CPU

conda install pytorch torchvision torchaudio cpuonly -c pytorch

GPU（CUDA 11.8）

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

四、验证安装

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

• True → GPU 可用
• False → 当前是 CPU 版

五、常见问题

❌ 下载慢

使用国内镜像：

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

❌ CUDA 不匹配

确保：

• 驱动版本 ≥ CUDA 要求
• PyTorch CUDA 版本 ≤ 驱动支持版本

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

• Ubuntu 版本
• 是否有 GPU
• 是否用 conda / 虚拟环境

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