温馨提示×

如何在centos装pytorch

小樊
41
2026-08-20 04:05:23
栏目: 智能运维

CentOS 上安装 PyTorch,推荐通过 Condapip 安装,下面给你几种常见且稳妥的方式(含 CPU / GPU)。


一、前置准备

1️⃣ 确认系统版本

cat /etc/centos-release

建议 CentOS 7 或 8,CentOS 6 已不推荐。


2️⃣ 安装 Python(建议 3.8–3.10)

CentOS 自带 Python 可能较老,建议用 pyenvconda 管理。


二、方式一(✅ 推荐):使用 Conda 安装 PyTorch

1️⃣ 安装 Miniconda

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

安装完成后重启终端或执行:

source ~/.bashrc

2️⃣ 创建虚拟环境

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

3️⃣ 安装 PyTorch(CPU 版本)

conda install pytorch torchvision torchaudio cpuonly -c pytorch

4️⃣ 安装 PyTorch(GPU 版本,CUDA 11.8 示例)

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

✅ 需要:

  • NVIDIA 显卡
  • 已安装对应版本驱动

三、方式二:使用 pip 安装(无 Conda)

1️⃣ 安装 pip

python3 -m pip install --upgrade pip

2️⃣ CPU 版本

pip install torch torchvision torchaudio

3️⃣ GPU 版本(CUDA 11.8)

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

四、验证安装是否成功

python
import torch

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

✅ 输出示例:

2.1.0
True   # 或 False(CPU)

五、常见问题

❓ 1. CentOS 报 GLIBC_2.17 not found

  • 升级系统
  • 或选择 较老版本 PyTorch

❓ 2. GPU 不可用

检查:

nvidia-smi

若没有输出,说明:

  • 未安装 NVIDIA 驱动
  • 或驱动版本不匹配

❓ 3. 网络慢

使用国内镜像:

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

六、推荐组合(稳定)

环境 推荐
新手 Conda + CPU
服务器 Conda + CUDA
生产 pip + 虚拟环境

如果你愿意,可以告诉我:

  • ✅ CentOS 版本
  • ✅ 是否有 GPU
  • ✅ 是否需要 CUDA

我可以给你 一条精确的安装命令

0