centos pytorch怎么训练

centos

小樊

2026-10-03 11:01:40

栏目: 智能运维

在 CentOS 上使用 PyTorch 进行训练，一般流程如下（以服务器/无 GUI 环境为例）：

一、准备环境（CentOS）

1. 更新系统

sudo yum update -y

2. 安装 Python（推荐 3.8+）

sudo yum install -y python3 python3-pip

查看版本：

python3 --version

二、安装 PyTorch

1. CPU 版本

pip3 install torch torchvision torchaudio

2. GPU 版本（需 NVIDIA 显卡 + 驱动）

先确认 CUDA 版本：

nvidia-smi

然后到官网选择对应命令，例如 CUDA 11.8：

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

三、验证 PyTorch

import torch
print(torch.__version__)
print(torch.cuda.is_available())  # GPU 应为 True

四、准备训练代码（示例）

示例：简单训练（CPU/GPU 自适应）

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

device = "cuda" if torch.cuda.is_available() else "cpu"

model = nn.Linear(10, 1).to(device)
optimizer = optim.SGD(model.parameters(), lr=0.01)
loss_fn = nn.MSELoss()

for epoch in range(100):
    x = torch.randn(32, 10).to(device)
    y = torch.randn(32, 1).to(device)

    pred = model(x)
    loss = loss_fn(pred, y)

    optimizer.zero_grad()
    loss.backward()
    optimizer.step()

    if epoch % 10 == 0:
        print(epoch, loss.item())

运行：

python3 train.py

五、后台训练（推荐服务器使用）

nohup python3 train.py > train.log 2>&1 &

查看日志：

tail -f train.log

六、常见问题

1. 缺少依赖

pip3 install numpy pandas matplotlib

2. 显存不足

减小 batch_size
使用 torch.cuda.empty_cache()

3. CentOS 老版本 Python

建议使用 conda 或 pyenv 管理 Python。

如果你有：

具体任务（分类 / 检测 / NLP）
是否有 GPU
数据格式

可以告诉我，我可以直接给你 完整训练示例。