1. 安装python和pip
首先，确认centos系统中已安装Python和pip。可使用以下命令完成安装：
sudo yum install python3 python3-pip
2. 创建虚拟环境（可选）
为了避免影响系统包，推荐创建一个独立的虚拟环境：
python3 -m venv myenv source myenv/bin/activate
3. 安装CUDA（可选）
若拥有NVIDIA显卡且需利用GPU加速，需安装CUDA Toolkit及cuDNN。首先，依据显卡型号和驱动版本，从NVIDIA官网下载对应版本的CUDA Toolkit并安装。接着，安装与CUDA版本相匹配的cuDNN库。
4. 安装PyTorch
根据系统配置（操作系统、Python版本、CUDA版本等），从PyTorch官网获取对应的安装指令。以下为一示例命令，用于在CPU环境下安装PyTorch：
pip3 install torch torchvision torchaudio -f https://download.pytorch.org/whl/cpu/torch_stable.html
若要使用GPU版本的PyTorch，可加入CUDA支持：
pip3 install torch torchvision torchaudio torchvision torchaudio -f https://download.pytorch.org/whl/cu113/torch_stable.html
请将cu113替换为所安装的CUDA版本。
5. 检查安装情况
安装完毕后，可通过以下命令验证PyTorch是否安装成功：
python -c "import torch; print(torch.__version__)"
如无错误提示，则表明PyTorch已成功安装。
6. 设置环境变量（可选）
为便于使用PyTorch，可将PyTorch的bin目录添加至系统PATH环境变量中：
echo 'export PATH=/path/to/pytorch/bin:$PATH' >> ~/.bashrc source ~/.bashrc
请用实际的PyTorch安装路径替代/path/to/pytorch。
7. 安装额外依赖（可选）
视项目需求，可能还需安装其他Python库，例如numpy、matplotlib等：
pip3 install numpy matplotlib
上述即为PyTorch在CentOS上的部署流程。如在安装时遇问题，可查阅PyTorch官方文档或相关社区资源。
© 版权声明
文章版权归作者所有，未经允许请勿转载。
THE END
