如何在CentOS上运行PyTorch

在CentOS上运行PyTorch，你需要先安装Python和pip，然后使用pip安装PyTorch

1. 更新系统软件包：

sudo yum update

2. 安装EPEL仓库（Extra Packages for Enterprise Linux）以获取额外的软件包：

sudo yum install epel-release

3. 安装开发工具和依赖项：

sudo yum groupinstall "Development Tools"
sudo yum install cmake3 git wget

4. 安装Python 3和pip：

sudo yum install python3 python3-pip

5. 使用pip安装PyTorch。根据你的系统和CUDA版本选择合适的命令。你可以在PyTorch官方网站找到适用于CentOS的命令：https://pytorch.org/get-started/locally/

例如，如果你想安装CPU版本的PyTorch，可以使用以下命令：

pip3 install torch torchvision torchaudio

如果你想安装支持CUDA的PyTorch版本，请根据你的CUDA版本选择合适的命令。例如，对于CUDA 11.1，可以使用以下命令：

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu111

6. 验证PyTorch安装成功。在终端中输入python3进入Python解释器，然后尝试导入PyTorch模块：

import torch
print(torch.__version__)

如果没有出现错误，并且显示了PyTorch的版本号，那么恭喜你，PyTorch已经成功安装在你的CentOS系统上了！