centos pytorch环境怎么建

centos

小樊

2026-09-18 13:00:43

栏目: 智能运维

在 CentOS 上搭建 PyTorch 环境，一般推荐用 Miniconda / Anaconda + 虚拟环境，最省心、兼容性最好。下面给你一套通用、稳妥的流程（CentOS 7 / 8 都适用）。

一、准备工作（CentOS）

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

三、创建 PyTorch 虚拟环境

1. 新建环境（示例 Python 3.10）

conda create -n pytorch python=3.10 -y

2. 激活环境

conda activate pytorch

四、安装 PyTorch（CPU / GPU）

✅ CPU 版本（通用）

pip install torch torchvision torchaudio

✅ GPU 版本（有 NVIDIA 显卡 + 驱动）

先确认：

nvidia-smi

然后根据 CUDA 版本安装，例如 CUDA 11.8：

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

官网选择工具：
https://pytorch.org/get-started/locally/

五、验证是否成功

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

True → GPU 可用
False → 当前为 CPU 版本

六、常见问题（CentOS 特有）

1. glibc 版本太低

CentOS 7 默认 glibc 2.17，某些新版 PyTorch 可能不支持
✅ 解决：

使用 conda 安装 torch

conda install pytorch torchvision torchaudio -c pytorch

2. 没有网络/内网环境

用离线 wheel
或用公司私有镜像源

七、推荐目录结构

~/envs/pytorch/
project/
 └── train.py

如果你告诉我：

✅ CentOS 版本（7 / 8 / Stream）
✅ 是否有 GPU
✅ 用途（训练 / 推理 / 服务器部署）

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