centos pytorch怎么配置

在 CentOS 上配置 PyTorch，一般分三步：系统环境准备 → 安装 Python / CUDA → 安装 PyTorch。下面按最常见、最稳妥的方式一步步来。

一、系统准备（CentOS 7 / 8 通用）

1️⃣ 更新系统

sudo yum update -y

2️⃣ 安装基础依赖

sudo yum install -y gcc gcc-c++ make \
    wget curl unzip git \
    epel-release

二、安装 Python（推荐 miniconda，最省事）

⚠️ CentOS 自带 Python 版本较老，不建议直接用系统 Python

1️⃣ 下载 Miniconda

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

安装完成后：

source ~/.bashrc

2️⃣ 创建 PyTorch 专用环境（推荐）

conda create -n torch python=3.10 -y
conda activate torch

三、安装 PyTorch（CPU / GPU 二选一）

✅ 方案一：CPU 版本（无显卡 / 服务器无 GPU）

conda install pytorch torchvision torchaudio cpuonly -c pytorch

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

1️⃣ 确认显卡驱动

nvidia-smi

如果没输出，先安装驱动（略复杂，可单独讲）

2️⃣ 安装 CUDA & cuDNN（conda 自动管理，推荐）

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

✅ 这是最省事的 GPU 安装方式，不需要手动装 CUDA

四、验证 PyTorch 是否安装成功

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

CPU 验证

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

GPU 验证

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

五、常见问题（CentOS 特有）

❓ 1. glibc 版本过低（CentOS 7）

ldd --version

如果 < 2.17，建议：

• 使用 conda 安装
• 或使用 Docker

❓ 2. 防火墙 / 离线环境

如果是内网服务器：

# 离线安装
conda install pytorch --offline

❓ 3. 多用户服务器

建议：

• 用 conda 环境
• 不要 sudo pip install

六、推荐组合（稳定）

七、我可以继续帮你

你可以直接告诉我：

1. ✅ CentOS 7 还是 8？
2. ✅ 有没有 NVIDIA 显卡？
3. ✅ 用 CPU 还是 GPU？
4. ✅ 是本地还是服务器？

我可以给你 完全定制的一套安装命令 ✅