linux下pytorch怎么安装

在 Linux 下安装 PyTorch，通常分为 CPU 版 和 GPU（CUDA）版，推荐用 conda 或 pip。下面给你最常用、最稳妥的方式。

一、安装前准备

1. 查看系统环境

uname -a
python3 --version

2. 推荐先装 Anaconda / Miniconda（可选但强烈建议）

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

创建环境：

conda create -n torch python=3.10
conda activate torch

二、CPU 版 PyTorch（最简单）

✅ pip 安装

pip install torch torchvision torchaudio

✅ conda 安装

conda install pytorch torchvision torchaudio cpuonly -c pytorch

三、GPU 版 PyTorch（CUDA）

1. 先确认 CUDA 驱动

nvidia-smi

看右上角 CUDA Version，比如 12.1

2. 安装对应版本（示例 CUDA 11.8）

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

或 conda：

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

⚠️ CUDA 版本必须 ≤ 驱动支持的版本

四、验证安装

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

输出：

• True → GPU 可用
• False → 当前是 CPU 版或未正确安装 CUDA

五、常见问题

❌ 速度慢 / 下载失败

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

❌ 多个 Python 混乱

用 which python 确认当前环境

如果你告诉我：

• ✅ Ubuntu / CentOS？
• ✅ 是否有 NVIDIA 显卡？
• ✅ Python 版本？

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