linux pytorch怎么安装

在 Linux 上安装 PyTorch，常见方式是使用 pip 或 conda。下面按场景给你最实用的方法。

一、使用 pip 安装（推荐，最简单）

1️⃣ 先确认 Python 版本

python3 --version

建议 Python ≥ 3.8

2️⃣ 创建虚拟环境（可选但强烈推荐）

python3 -m venv pytorch_env
source pytorch_env/bin/activate

3️⃣ 安装 CPU 版 PyTorch

pip install torch torchvision torchaudio

4️⃣ 安装 GPU 版（CUDA，示例 CUDA 11.8）

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

如果你是 RTX 30/40 系列，一般用 cu118 或 cu121

二、使用 conda 安装（适合科研/多环境）

1️⃣ 安装 Miniconda（如未安装）

wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
bash Miniconda3-latest-Linux-x86_64.sh

2️⃣ 创建环境

conda create -n torch python=3.10
conda activate torch

3️⃣ 安装 PyTorch

CPU：

conda install pytorch torchvision torchaudio cpuonly -c pytorch

GPU（CUDA 11.8）：

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

三、验证是否安装成功

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

GPU 检查：

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

输出 True 说明 GPU 可用 ✅

四、常见问题

❓ 找不到 CUDA

• 先确认驱动：nvidia-smi
• 驱动版本要 ≥ CUDA 要求

❓ pip 太慢

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

如果你告诉我：

• ✅ 是否有 NVIDIA 显卡
• ✅ Ubuntu / CentOS / 其他
• ✅ 是否用服务器/容器

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