Ubuntu上PyTorch安装步骤详解

以下是在Ubuntu上安装PyTorch的详细步骤：

准备工作

• 更新系统包列表：sudo apt update。
• 安装Python和pip：sudo apt install python3 python3-pip python3-venv -y。

安装PyTorch

使用pip安装

• CPU版本：pip3 install torch==2.6.0 torchvision==0.17.0 torchaudio==2.6.0 --index-url https://download.pytorch.org/whl/cpu。
• GPU版本：先安装对应CUDA版本，再执行pip3 install torch==2.6.0 torchvision==0.17.0 torchaudio==2.6.0 --index-url https://download.pytorch.org/whl/cu126（以CUDA 12.6为例）。

使用conda安装

• 安装Anaconda：从官网下载安装脚本并执行，如bash Anaconda3-2023.07-Linux-x86_64.sh，安装后激活环境source ~/.bashrc。
• 创建虚拟环境：conda create -n pytorch_env python=3.8，激活环境conda activate pytorch_env。
• 安装PyTorch：CPU版本用conda install pytorch torchvision torchaudio cpuonly -c pytorch；GPU版本用conda install pytorch torchvision torchaudio cudatoolkit=<cuda_version> -c pytorch（将<cuda_version>替换为实际版本）。

验证安装

在Python解释器中运行：

import torch
print(torch.__version__)
print(torch.cuda.is_available())