centos pytorch环境怎么搭

在 CentOS 上搭建 PyTorch 环境，通常推荐用 conda（Anaconda/Miniconda），简单、稳定，也方便管理 CUDA 版本。下面给你一套通用、可落地的流程（CentOS 7 / 8 都适用）。

一、准备工作

1. 更新系统

sudo yum update -y

2. 安装基础依赖

sudo yum install -y gcc gcc-c++ make wget bzip2

二、安装 Miniconda（推荐）

1. 下载 Miniconda

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

2. 安装

bash Miniconda3-latest-Linux-x86_64.sh

一路回车，最后输入 yes 初始化。

3. 生效环境

source ~/.bashrc

验证：

conda --version

三、创建 PyTorch 虚拟环境

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

四、安装 PyTorch（重点）

✅ 有 NVIDIA GPU（推荐）

查看 CUDA 驱动支持

nvidia-smi

右上角会显示 最高支持 CUDA 版本

安装 PyTorch（示例 CUDA 11.8）

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

如果是 CUDA 12.x，把 11.8 改成 12.1

✅ 只有 CPU（服务器/测试环境）

conda install pytorch torchvision torchaudio cpuonly -c pytorch

五、验证 PyTorch 是否安装成功

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

检查 GPU 是否可用

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

输出 True 即成功 ✅

六、常见问题（CentOS 特有）

1. GLIBC 版本过低

CentOS 7 可能遇到：

ImportError: /lib64/libc.so.6: version 'GLIBC_2.17' not found

✅ 解决：

• 使用 conda 安装 PyTorch（不要 pip）
• 或升级系统到 CentOS 8 / Rocky / AlmaLinux

2. pip 安装失败

不建议在 CentOS 直接用 pip 装 PyTorch，容易踩坑。

七、推荐组合（稳定）

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

• ✅ CentOS 版本
• ✅ 是否有 GPU
• ✅ 是否需要训练 / 推理 / 服务器部署

我可以直接给你一条命令版配置。