以下是在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。
以上就是关于“CentOS服务器上PyTorch部署指南”的相关介绍,筋斗云是国内较早的云主机应用的服务商,拥有10余年行业经验,提供丰富的云服务器、租用服务器等相关产品服务。云服务器资源弹性伸缩,主机vCPU、内存性能强悍、超高I/O速度、故障秒级恢复;电子化备案,提交快速,专业团队7×24小时服务支持!
简单好用、高性价比云服务器租用链接:https://www.jindouyun.cn/product/cvm