ubuntu pytorch如何配置

ubuntu

小樊

2026-09-23 02:32:09

栏目: 智能运维

下面按在 Ubuntu 上配置 PyTorch的常见场景，给你一份清晰、可操作的指南（以 Ubuntu 20.04 / 22.04 为例，其他版本类似）。

一、前置准备

1. 更新系统

sudo apt update
sudo apt upgrade -y

2. 安装 Python 与 pip

Ubuntu 一般自带 Python3：

python3 --version
pip3 --version

如未安装：

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

二、推荐：使用虚拟环境（很重要）

避免污染系统 Python：

python3 -m venv pytorch_env
source pytorch_env/bin/activate

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

三、安装 PyTorch（CPU / GPU）

✅ 方案 1：仅 CPU（最简单）

pip install torch torchvision torchaudio

✅ 方案 2：NVIDIA GPU（CUDA）

1️⃣ 确认显卡和驱动

nvidia-smi

能看到 CUDA Version 说明驱动正常。

2️⃣ 安装对应 PyTorch（示例 CUDA 11.8）

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

官方选择地址（强烈推荐）： https://pytorch.org/get-started/locally/

四、验证 PyTorch 是否安装成功

python

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

五、常见问题

❓ pip 太慢

使用国内镜像：

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

❓ 多 CUDA 版本冲突

不需要手动装 CUDA Toolkit（PyTorch 自带）
只需正确 NVIDIA 驱动

❓ Jupyter 使用

pip install jupyter
jupyter notebook

六、最小示例

import torch

x = torch.rand(3, 3)
print(x)
print("CUDA:", torch.cuda.is_available())

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

Ubuntu 版本
是否有 NVIDIA 显卡
是否用 Docker / Conda

我可以给你完全定制化的安装命令。