centos怎么测试pytorch

在 CentOS 上测试 PyTorch 通常分两种情况：
1️⃣ 已安装 PyTorch，想验证是否能用
2️⃣ 还没装，想从零开始装并测试

下面一步步来。

一、确认环境（CentOS）

cat /etc/centos-release
python3 --version

推荐：

• CentOS 7 / 8
• Python 3.8+

二、安装 PyTorch（以 CPU 为例）

1️⃣ 创建虚拟环境（推荐）

python3 -m venv pytorch_env
source pytorch_env/bin/activate

2️⃣ 安装 PyTorch（CPU 版）

pip install torch torchvision torchaudio

如果有 NVIDIA GPU，去官网选 CUDA 版本： https://pytorch.org/get-started/locally/

例如：

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

三、测试 PyTorch 是否安装成功

✅ 基础测试

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

输出示例：

2.1.0

✅ 测试张量计算

import torch

x = torch.rand(3, 3)
y = torch.rand(3, 3)
print(x + y)

✅ 测试 GPU（如有）

import torch

print(torch.cuda.is_available())

• True → GPU 可用
• False → 只用 CPU

✅ 简单神经网络测试

import torch
import torch.nn as nn

model = nn.Linear(10, 1)
x = torch.rand(5, 10)
out = model(x)
print(out.shape)

四、常见问题（CentOS）

1️⃣ 缺少依赖

sudo yum install -y python3-devel gcc

2️⃣ GLIBC 版本太低（CentOS 7）

• 老系统可能不支持新 PyTorch
• 解决：
  • 使用旧版 PyTorch
  • 或用 conda

3️⃣ 推荐用 conda（最稳）

conda create -n torch python=3.9
conda activate torch
conda install pytorch torchvision torchaudio cpuonly -c pytorch

五、总结

✅ 测试核心就一句：

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

如果你告诉我：

• CentOS 版本
• 是否用 GPU
• Python 版本

我可以给你完全匹配的安装命令。