如何在ubuntu装pyorch

在 Ubuntu 上安装 PyTorch，通常推荐用 conda 或 pip，并根据你的硬件（是否有 NVIDIA GPU）选择 CPU 或 CUDA 版本。下面给你几种常见方式。

一、先确认环境

1. 查看 Ubuntu 版本

lsb_release -a

2. 查看是否有 NVIDIA GPU（可选）

nvidia-smi

• 有输出 → 可用 CUDA 版
• 无输出 → 用 CPU 版

二、使用 pip 安装（最简单）

1. 安装 Python 和 pip

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

2. 创建虚拟环境（推荐）

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

3. 安装 PyTorch

CPU 版本

pip install torch torchvision torchaudio

GPU（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

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

五、常见问题

• pip 很慢：用清华源

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

• CUDA 不匹配：以 nvidia-smi 显示的 CUDA 版本为准（向下兼容）

如果你告诉我：

• Ubuntu 版本
• 是否有 GPU
• 用 pip 还是 conda

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