CentOS系统PyTorch安装详细教程
一、前置准备
1. 更新系统包
确保系统软件包为最新版本，避免兼容性问题：
sudo yum update -y
2. 安装必要依赖
PyTorch及后续工具链需要编译工具、Python环境和网络工具支持：
sudo yum install -y python3 python3-pip python3-devel gcc openssl-devel bzip2-devel libffi-devel
3. 检查NVIDIA GPU（可选，GPU加速必备）
若需使用GPU加速，需确认显卡支持CUDA并安装驱动：
nvidia-smi # 查看显卡型号及驱动版本（需显示CUDA版本，如12.2）
注：驱动版本需满足PyTorch对应CUDA版本的要求（如PyTorch 2.1需CUDA 11.8+）。
二、选择安装方式
PyTorch提供pip（官方推荐）和conda（适合环境管理）两种安装方式，以下分别说明。
方式一：使用pip安装（适合纯Python环境）
1. 安装pip（若未安装）
sudo yum install -y python3-pip
2. 选择PyTorch版本
根据是否需要GPU加速选择安装命令：
- CPU版本（无GPU支持）：
pip3 install torch torchvision torchaudio - GPU版本（需匹配CUDA版本，如CUDA 11.7）：
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117注：CUDA版本需与显卡驱动匹配（如
nvidia-smi显示CUDA 11.7，则选择cu117）。
方式二：使用conda安装（适合环境隔离）
1. 安装Miniconda（轻量级conda发行版）
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
bash Miniconda3-latest-Linux-x86_64.sh # 按提示完成安装（建议选择"Just Me"）
2. 创建并激活conda环境（隔离依赖）
conda create -n pytorch_env python=3.8 # 推荐Python 3.7-3.8（PyTorch官方最佳兼容版本）
conda activate pytorch_env
3. 安装PyTorch
- CPU版本：
conda install pytorch torchvision torchaudio cpuonly -c pytorch - GPU版本（需匹配CUDA版本，如CUDA 11.7）：
conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch -c conda-forge注：conda会自动解决依赖冲突，无需手动安装CUDA Toolkit（但需提前安装显卡驱动）。
三、验证安装
安装完成后，通过Python代码验证PyTorch是否成功安装及GPU可用性：
import torch
print("PyTorch版本:", torch.__version__) # 输出版本号（如2.1.0）
print("CUDA可用性:", torch.cuda.is_available()) # GPU版本应返回True，CPU版本返回False
若torch.cuda.is_available()返回True，则表示GPU加速功能正常。
四、常见问题解决
1. 依赖冲突
若安装时出现依赖冲突（如numpy版本不兼容），可降级冲突包：
pip install "numpy<1.24.0" # 例如降级numpy到1.23.x
2. nvcc命令找不到
若使用GPU版本时报错nvcc: command not found，需检查CUDA环境变量是否配置：
echo 'export PATH=/usr/local/cuda/bin:$PATH' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc
source ~/.bashrc
3. 版本兼容性
- PyTorch 2.1需CUDA 11.8+，PyTorch 1.13需CUDA 11.6+，需确保显卡驱动版本满足要求。
通过以上步骤，即可在CentOS系统上成功安装PyTorch，并根据需求选择CPU或GPU版本。安装过程中若遇问题，建议参考PyTorch官方文档或社区论坛（如Stack Overflow）获取进一步帮助。
以上就是关于“CentOS上PyTorch的安装教程来了”的相关介绍，筋斗云是国内较早的云主机应用的服务商，拥有10余年行业经验，提供丰富的云服务器、租用服务器等相关产品服务。云服务器资源弹性伸缩，主机vCPU、内存性能强悍、超高I/O速度、故障秒级恢复；电子化备案，提交快速，专业团队7×24小时服务支持！
简单好用、高性价比云服务器租用链接：https://www.jindouyun.cn/product/cvm