ubuntu怎么安装pytorch

ubuntu

小樊

2026-09-23 02:31:06

栏目: 智能运维

在 Ubuntu 上安装 PyTorch，通常步骤如下（以 CPU / NVIDIA GPU + CUDA 为例）：

一、准备工作

Ubuntu 一般自带 Python3，检查：

python3 --version
pip3 --version

如果没有 pip：

sudo apt update
sudo apt install python3-pip

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

pip install torch torchvision torchaudio

nvidia-smi

能看到 CUDA Version 表示驱动已装好。

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

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

python

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

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

直接用 pip 即可，不推荐用 apt 装 PyTorch。

如果你告诉我：

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