centos pytorch怎么导入

在 CentOS 上“导入 PyTorch”通常分两步：安装 PyTorch 和 在 Python 中 import。下面按常见情况说明。

一、确认环境（CentOS）

cat /etc/centos-release
python3 --version
pip3 --version

如果没有 Python3：

sudo yum install -y python3 python3-pip

二、安装 PyTorch（CentOS）

1️⃣ CPU 版本（最常用，服务器无 GPU）

pip3 install torch torchvision torchaudio

2️⃣ GPU 版本（有 NVIDIA 显卡 + CUDA）

先确认 CUDA 版本：

nvidia-smi

例如 CUDA 11.8：

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

三、导入 PyTorch（关键步骤）

进入 Python：

python3

然后输入：

import torch
print(torch.__version__)

如果没有报错，说明导入成功 ✅

四、常见问题（CentOS 特有）

❌ 1. ImportError: libgomp.so.1

sudo yum install -y libgomp

❌ 2. GLIBC 版本过低（老 CentOS 7）

建议：

• 使用 conda 安装（推荐）

conda install pytorch torchvision torchaudio cpuonly -c pytorch

❌ 3. pip 版本太旧

pip3 install --upgrade pip

五、推荐方案（CentOS 服务器）

# 使用 conda 最稳
conda create -n torch python=3.9
conda activate torch
conda install pytorch torchvision torchaudio cpuonly -c pytorch
python
>>> import torch

如果你能告诉我：

• CentOS 版本（7 / 8 / Stream）
• 是否 GPU
• Python 版本

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