centos pytorch如何导入

在 CentOS 上“导入 PyTorch”通常包含两步：

1. 安装 PyTorch（CentOS 环境）
2. 在 Python 中 import torch

下面按常见情况说明。

一、确认基础环境（CentOS）

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

CentOS 7 一般自带 Python 2，需要自己装 Python 3：

yum install -y python3 python3-pip

二、安装 PyTorch（CPU 版，最常用）

1️⃣ 使用 pip 安装（推荐）

pip3 install torch torchvision torchaudio

验证是否安装成功：

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

2️⃣ 使用官方指定命令（更稳妥）

去 PyTorch 官网选：

• OS: Linux
• Package: pip
• Language: Python
• Compute: CPU

示例（CPU）：

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

三、GPU 版（CentOS + NVIDIA）

前提

• 已安装 NVIDIA 驱动
• 已安装 CUDA

查看 CUDA：

nvidia-smi
nvcc --version

安装 GPU 版 PyTorch

例如 CUDA 11.8：

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

验证 GPU：

import torch
print(torch.cuda.is_available())

四、在代码中“导入”PyTorch

import torch
import torch.nn as nn
import torch.optim as optim

x = torch.tensor([1.0, 2.0, 3.0])
print(x)

五、常见问题（CentOS 特有）

❌ 找不到 libgomp / GLIBCXX

yum install -y libgomp gcc

❌ ImportError: libstdc++.so.6

yum install -y centos-release-scl
yum install -y devtoolset-9
scl enable devtoolset-9 bash

❌ pip 版本太低

pip3 install --upgrade pip

如果你能告诉我：

• CentOS 版本（7 / 8 / Stream）
• 是否用 GPU
• 是否需要 conda / 虚拟环境

我可以给你完全贴合你机器的安装命令。