CentOS服务器上PyTorch部署指南

以下是在CentOS服务器上部署PyTorch的指南：

系统更新与依赖安装

更新系统：sudo yum update -y。安装基础依赖：sudo yum groupinstall -y "Development Tools" sudo yum install -y python3 python3-devel python3-pip。

虚拟环境创建（可选）

使用conda创建：先安装Anaconda，然后conda create -n pytorch_env python=3.8，再conda activate pytorch_env。使用virtualenv创建：sudo yum install -y python3-virtualenv，接着virtualenv pytorch_env，最后source pytorch_env/bin/activate。

PyTorch安装

若用CPU版本，执行pip3 install torch torchvision torchaudio。若用GPU版本，先安装对应CUDA和cuDNN，再执行pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu版本号，如pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117。

验证安装

在Python交互式环境中输入import torch，print(torch.__version__)，print(torch.cuda.is_available())，若版本号正常显示且GPU版本返回True，则安装成功。

配置为后台服务（可选）

创建服务文件/etc/systemd/system/pytorch.service，写入内容[Unit] Description=PyTorch Application Service [Service] ExecStart=/path/to/pytorch_env/bin/python /path/to/your_application.py WorkingDirectory=/path/to/your_application_directory User=your_username Group=your_groupname Restart=always [Install] WantedBy=multi-user.target。然后执行sudo systemctl start pytorch和sudo systemctl enable pytorch。