linux下pytorch怎么配置

在 Linux 下配置 PyTorch，通常分为系统环境准备 → 安装 Python/依赖 → 安装 PyTorch → 验证几个步骤。下面给你一份通用且常用的做法（以 Ubuntu 为例）。

一、系统环境准备

1. 更新系统

sudo apt update
sudo apt upgrade -y

2. 安装基础依赖

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

二、使用虚拟环境（推荐）

避免污染系统 Python：

python3 -m venv pytorch_env
source pytorch_env/bin/activate

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

三、安装 PyTorch

✅ 方式一：CPU 版本（无 GPU）

pip install torch torchvision torchaudio

✅ 方式二：GPU 版本（NVIDIA CUDA）

1. 先确认显卡和驱动

nvidia-smi

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

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

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

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

四、验证安装

python -c "import torch; print(torch.__version__)"

GPU 验证：

python -c "import torch; print(torch.cuda.is_available())"

返回 True 表示 GPU 可用。

五、常见问题

1. pip 太慢

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

2. 多个 Python 版本

用 python3 和 pip3 明确指定

3. conda 用户（可选）

conda create -n torch python=3.10
conda activate torch
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

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

• 是否用 GPU
• Linux 发行版（Ubuntu / CentOS / 服务器）
• 是否用 conda

我可以直接给你精确命令。