温馨提示×

pytorch在centos咋启动

小樊
34
2026-10-18 02:17:08
栏目: 智能运维

在 CentOS 上“启动 PyTorch”通常不是像启动一个服务那样,而是安装并运行 PyTorch 程序。下面按常见使用场景给你说明。


一、确认系统环境(CentOS)

先确认 CentOS 版本和 Python 环境:

cat /etc/centos-release
python3 --version

一般建议:

  • CentOS 7 / 8
  • Python 3.8+

如果没有 Python3:

sudo yum install python3 python3-pip

二、创建虚拟环境(推荐)

python3 -m venv pytorch_env
source pytorch_env/bin/activate

三、安装 PyTorch(CentOS)

1️⃣ CPU 版本(最常用)

pip install torch torchvision torchaudio

2️⃣ GPU 版本(有 NVIDIA 显卡 + CUDA)

先确认驱动和 CUDA:

nvidia-smi

然后去官网选命令,例如 CUDA 11.8:

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

四、验证 PyTorch 是否可用

python

在 Python 里输入:

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

输出类似:

2.1.0
True   # 或 False(CPU 版)

五、运行你的 PyTorch 程序

假设你有 train.py:

python train.py

如果是训练任务,建议后台运行:

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

查看日志:

tail -f train.log

六、如果你说的“启动”是指服务(如推理 API)

可以用:

  • Flask / FastAPI
  • gunicorn 启动

示例:

pip install fastapi uvicorn
uvicorn main:app --host 0.0.0.0 --port 8000

七、常见问题

❌ 缺依赖

sudo yum install -y glibc glib2 libgomp

❌ pip 太老

pip install --upgrade pip

如果你能具体说:

  • 是 CPU 还是 GPU
  • 是 训练 / 推理 / 服务
  • 报错信息是什么

我可以直接帮你写完整启动命令。

0 踩