centos下pytorch怎么用

在 CentOS 下使用 PyTorch，一般流程是：安装 Python 环境 → 安装 PyTorch → 验证 → 写代码运行。下面按最常见场景给你一份实用指南（含 GPU 情况）。

一、确认系统环境

cat /etc/centos-release
python3 --version
gcc --version

CentOS 7 常见，Python 默认可能是 2.7，建议用 Python 3.8+

二、安装 Python（如未安装）

CentOS 7

sudo yum install -y python3 python3-devel

CentOS 8 / Stream

sudo dnf install -y python3 python3-devel

三、创建虚拟环境（推荐）

python3 -m venv pytorch_env
source pytorch_env/bin/activate

升级 pip：

pip install --upgrade pip

四、安装 PyTorch

1️⃣ CPU 版本（最省事）

pip install torch torchvision torchaudio

2️⃣ GPU 版本（CUDA）

先确认 CUDA 版本：

nvidia-smi

例如 CUDA 11.8：

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

⚠️ CentOS 老系统可能缺少新版 GLIBC，GPU 版容易踩坑

五、验证 PyTorch

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

输出：

2.x.x
True   # 或 False（CPU）

六、简单示例

import torch

x = torch.randn(3, 3)
y = torch.randn(3, 3)

print(x + y)
print(torch.cuda.is_available())

运行：

python test.py

七、常见问题（CentOS 特有）

1️⃣ GLIBC 版本过低

报错：version 'GLIBC_2.17' not found

✅ 解决：

• 用 CPU 版
• 或升级系统（不推荐生产机）
• 或用 conda

2️⃣ 推荐用 Conda（最稳）

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.9
conda activate torch
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

八、总结建议

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

• ✅ CentOS 版本
• ✅ 是否有 GPU
• ✅ 用于训练还是推理

我可以给你完全定制的安装命令。