ubuntu pytorch怎么安装

在 Ubuntu 上安装 PyTorch 通常推荐用 Conda 或 pip，并且要根据你是否有 NVIDIA 显卡（CUDA） 来选择版本。下面分情况说明。

一、先确认环境

1. 查看 Ubuntu 版本

lsb_release -a

2. 查看是否有 NVIDIA 显卡

nvidia-smi

• 有输出 → 支持 CUDA
• 没输出 → 只能装 CPU 版

二、使用 pip 安装（最常用）

✅ CPU 版本

pip install torch torchvision torchaudio

✅ CUDA 版本（以 CUDA 11.8 为例）

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

其他 CUDA 版本可去官网选：
https://pytorch.org/get-started/locally/

三、使用 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

四、验证安装

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

查看是否支持 GPU：

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

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

五、常见问题

1. pip 太慢

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

2. 多个 Python 版本

建议用 虚拟环境（venv 或 conda）

3. 安装后 import 报错

• 检查 Python 版本（PyTorch 不支持太老的 Python）
• 检查是否装错 CPU / CUDA 版本

如果你告诉我：

• Ubuntu 版本
• 是否有 NVIDIA 显卡
• 用 pip 还是 conda

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