ubuntu pytorch咋配置环境

ubuntu

小樊

2026-10-19 11:17:36

栏目: 智能运维

下面给你一份在 Ubuntu 上配置 PyTorch 环境的常用、稳妥方案（CPU / GPU 都覆盖），按步骤来即可。

一、准备工作

1. 更新系统

sudo apt update
sudo apt upgrade -y

2. 安装基础依赖

sudo apt install -y python3 python3-pip python3-venv

二、创建 Python 虚拟环境（推荐）

避免污染系统环境：

mkdir ~/pytorch_env
cd ~/pytorch_env
python3 -m venv venv
source venv/bin/activate

激活后命令行前面会出现 (venv)。

三、安装 PyTorch

✅ CPU 版本（最简单）

pip install torch torchvision torchaudio

✅ GPU 版本（NVIDIA 显卡）

1️⃣ 先确认显卡驱动

nvidia-smi

能看到显卡信息说明驱动已装。

2️⃣ 安装 CUDA（以 CUDA 11.8 为例）

sudo apt install nvidia-cuda-toolkit

或参考 NVIDIA 官方 run 文件安装（更推荐）。

3️⃣ 安装对应 PyTorch

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

版本号（cu118 / cu121）要和你显卡驱动支持的 CUDA 匹配

四、验证是否安装成功

python

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

五、常用问题

1. pip 很慢

pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

2. Jupyter 使用

pip install jupyter
jupyter notebook

3. 退出环境

deactivate

六、推荐组合（新手）

Ubuntu 22.04
Python 3.10
CUDA 11.8
PyTorch 2.x

如果你告诉我：

Ubuntu 版本
是否有 NVIDIA 显卡
是否用 Anaconda

我可以直接给你一条最合适的具体命令。