centos pytorch怎么测试

在 CentOS 上测试 PyTorch 是否安装成功、能否正常使用（尤其是 GPU），可以按下面步骤来。

一、基础环境确认

1. 查看系统

cat /etc/centos-release

2. 查看 Python 版本

python3 --version

建议使用 Python 3.8 / 3.9 / 3.10（PyTorch 官方支持较好）

二、进入 Python 测试 PyTorch

1. 进入 Python

python3

2. 导入 PyTorch

import torch
print(torch.__version__)

✅ 能正常输出版本号说明安装成功。

三、CPU 测试

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

如果没有报错，说明 CPU 版本可用。

四、GPU（CUDA）测试（非常重要）

1. 查看 CUDA 是否可用

torch.cuda.is_available()

• 返回 True → GPU 可用
• 返回 False → 未使用 GPU（可能是驱动/CUDA/PyTorch 版本不匹配）

2. 查看 GPU 数量和设备名

print(torch.cuda.device_count())
print(torch.cuda.get_device_name(0))

3. 实际 GPU 运算测试

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
x = torch.rand(1000, 1000).to(device)
y = torch.rand(1000, 1000).to(device)
z = x @ y
print(z.device)

五、常见问题（CentOS 特有）

1. libstdc++.so.6 版本过低

报错类似：

GLIBCXX_3.4.29 not found

解决：

strings /usr/lib64/libstdc++.so.6 | grep GLIBCXX

通常需升级 gcc 或使用 conda 环境。

2. CentOS 7 默认 Python 太老

✅ 强烈建议使用 conda：

conda create -n torch python=3.9
conda activate torch
pip install torch torchvision

3. 显卡驱动问题

nvidia-smi

• 能看到 GPU → 驱动正常
• 看不到 → 先装 NVIDIA 驱动

六、快速总结（最常用）

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

如果你愿意，可以告诉我：

• CentOS 版本（7 / 8 / Stream）
• 是否用 conda
• 是否要用 GPU

我可以给你 一条最适合你的安装+测试命令。