怎样在ubuntu配pytorch

在 Ubuntu 上配置 PyTorch，一般步骤如下（以 Ubuntu 20.04 / 22.04 为例）：

一、准备工作

1. 更新系统

sudo apt update
sudo apt upgrade -y

2. 安装 Python 和 pip

Ubuntu 通常自带 Python3：

python3 --version

安装 pip：

sudo apt install python3-pip -y

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

避免污染系统环境：

sudo apt install python3-venv -y
python3 -m venv pytorch-env
source pytorch-env/bin/activate

二、安装 PyTorch

✅ 方式一：CPU 版本（无显卡）

pip install torch torchvision torchaudio

✅ 方式二：GPU 版本（有 NVIDIA 显卡）

1. 安装 NVIDIA 驱动

ubuntu-drivers devices
sudo ubuntu-drivers autoinstall
reboot

验证：

nvidia-smi

2. 安装 CUDA（以 CUDA 11.8 为例）

sudo apt install nvidia-cuda-toolkit -y

或参考官方： https://developer.nvidia.com/cuda-toolkit

3. 安装对应 PyTorch（示例 CUDA 11.8）

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

三、验证安装

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

多个 Python 版本

明确使用：

python3 -m pip install torch

五、最简总结（CPU）

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

如果你告诉我：

• Ubuntu 版本
• 是否有 NVIDIA 显卡
• 是否需要 Jupyter / Conda

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