linux pytorch如何配置

下面按在 Linux 上配置 PyTorch的常见场景，给你一套清晰、可操作的步骤。假设你使用的是 Ubuntu / CentOS / Debian 等常见发行版。

一、准备工作（必做）

1. 更新系统

# Ubuntu / Debian
sudo apt update && sudo apt upgrade -y

# CentOS
sudo yum update -y

2. 安装 Python 与 pip

推荐 Python 3.8–3.11

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

# CentOS
sudo yum install python3 python3-pip -y

查看版本：

python3 --version
pip3 --version

二、创建虚拟环境（强烈推荐）

mkdir pytorch_env && cd pytorch_env
python3 -m venv venv
source venv/bin/activate

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

三、安装 PyTorch（核心）

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

pip install torch torchvision torchaudio

✅ 方式二：GPU 版本（有 NVIDIA 显卡）

1️⃣ 安装 NVIDIA 驱动

nvidia-smi

能看到显卡信息说明驱动已装好。

2️⃣ 安装 CUDA（以 CUDA 11.8 为例）

sudo apt install nvidia-cuda-toolkit
nvcc --version

3️⃣ 安装对应 PyTorch

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

其他 CUDA 版本（如 cu121）请参考官网： https://pytorch.org/get-started/locally/

四、验证 PyTorch 是否安装成功

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

✅ GPU 验证：

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

输出 True 表示 GPU 可用。

五、常见问题

1. pip 下载慢

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

2. 多用户服务器

• 用 venv 或 conda
• 不要使用 sudo pip

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

六、最小测试代码

import torch

x = torch.rand(3, 3)
print(x)
print("CUDA:", torch.cuda.is_available())

如果你能告诉我：

• Linux 发行版
• 是否有 NVIDIA 显卡
• 是否用服务器 / 容器 / WSL

我可以给你更精确的配置命令。