温馨提示×

centos怎么安装pytorch

小樊
46
2026-09-09 10:27:50
栏目: 智能运维

CentOS 上安装 PyTorch,一般有两种常见方式:pip 安装(最常用、最简单)和 conda 安装。下面我按推荐程度给你说明。


一、使用 pip 安装(推荐,适合大多数情况)

1. 安装 Python 和 pip

CentOS 7/8 默认有 Python,但版本可能较低,建议先确认:

python3 --version
pip3 --version

如果没有:

sudo yum install python3 python3-pip -y

2. 升级 pip(强烈建议)

pip3 install --upgrade pip

3. 安装 PyTorch(CPU 版)

pip3 install torch torchvision torchaudio

4. 安装 PyTorch(GPU 版,CUDA 11.8 示例)

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

其他 CUDA 版本可去官网查看:
https://pytorch.org/get-started/locally/

5. 验证安装

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

二、使用 conda 安装(适合数据科学/多环境管理)

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 pytorch python=3.10
conda activate pytorch

3. 安装 PyTorch

CPU:

conda install pytorch torchvision torchaudio cpuonly -c pytorch

GPU(CUDA 11.8):

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

三、常见问题

1. CentOS 老版本 glibc 太低

PyTorch 新版可能要求 glibc ≥ 2.17(CentOS 7 满足),但 CentOS 6 不支持

2. 缺少依赖

sudo yum install -y libgomp

3. 国内加速(可选)

pip3 config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

四、快速总结

新手 / 服务器:pip 安装
多项目 / 环境隔离:conda 安装
有 NVIDIA 显卡:选 CUDA 版本

如果你能告诉我:

  • CentOS 版本(7 / 8 / Stream)
  • 是否有 GPU
  • Python 版本

我可以给你精确的命令

0