在 CentOS 上“启动 PyTorch”通常不是像启动一个服务那样,而是安装并运行 PyTorch 程序。下面按常见使用场景给你说明。
先确认 CentOS 版本和 Python 环境:
cat /etc/centos-release
python3 --version
一般建议:
如果没有 Python3:
sudo yum install python3 python3-pip
python3 -m venv pytorch_env
source pytorch_env/bin/activate
pip install torch torchvision torchaudio
先确认驱动和 CUDA:
nvidia-smi
然后去官网选命令,例如 CUDA 11.8:
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
python
在 Python 里输入:
import torch
print(torch.__version__)
print(torch.cuda.is_available())
输出类似:
2.1.0
True # 或 False(CPU 版)
假设你有 train.py:
python train.py
如果是训练任务,建议后台运行:
nohup python train.py > train.log 2>&1 &
查看日志:
tail -f train.log
可以用:
gunicorn 启动示例:
pip install fastapi uvicorn
uvicorn main:app --host 0.0.0.0 --port 8000
sudo yum install -y glibc glib2 libgomp
pip install --upgrade pip
如果你能具体说:
我可以直接帮你写完整启动命令。