在centos系统上搭建pytorch开发环境，可以按照以下步骤进行：
1. 更新系统
首先，确保你的CentOS系统是最新的。打开终端并运行以下命令：
sudo yum update -y
2. 安装依赖包
安装一些必要的依赖包，包括编译工具和Python开发库：
sudo yum groupinstall -y "Development Tools" sudo yum install -y python3 python3-devel numpy
3. 安装Python虚拟环境
使用venv模块来创建一个Python虚拟环境：
python3 -m venv pytorch_env source pytorch_env/bin/activate
4. 安装PyTorch
根据你的硬件配置（CPU或GPU）选择合适的PyTorch版本进行安装。以下是一些常见的安装命令：
CPU版本
pip install torch torchvision torchaudio
GPU版本（CUDA 11.7）
如果你有NVIDIA GPU并且已经安装了CUDA 11.7，可以使用以下命令安装PyTorch：
pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117
如果你使用的是其他版本的CUDA，请参考PyTorch官网获取相应的安装命令。
5. 验证安装
安装完成后，可以通过以下命令验证PyTorch是否安装成功：
import torch print(torch.__version__) print(torch.cuda.is_available()) # 如果是GPU版本，应该返回True
6. 安装其他常用库
根据你的项目需求，可能还需要安装其他常用的Python库，例如：
pip install matplotlib pandas scikit-learn
7. 配置IDE
如果你使用的是IDE（如PyCharm），可以将项目的解释器设置为刚刚创建的虚拟环境。
在PyCharm中设置解释器：
- 打开PyCharm，进入File -> Settings（或PyCharm -> Preferences）。
- 选择Project:
-> Python Interpreter。 - 点击齿轮图标，选择Add...。
- 选择Existing environment，然后选择你创建的虚拟环境的解释器路径（通常是pytorch_env/bin/python）。
- 点击OK完成设置。
通过以上步骤，你就可以在CentOS系统上成功搭建PyTorch开发环境了。
