在CentOS上安装PyTorch需要一些依赖项，包括Python、pip、CUDA和cuDNN等
sudo yum update -y
sudo yum install python3 -y
sudo yum install python3-pip -y
安装CUDA Toolkit。根据您的CentOS版本和GPU型号，请从NVIDIA官方网站下载并安装合适的CUDA Toolkit： https://developer.nvidia.com/cuda-toolkit-archive
安装cuDNN。同样地，请根据您的CentOS版本和GPU型号，从NVIDIA官方网站下载并安装合适的cuDNN库： https://developer.nvidia.com/cudnn
下载PyTorch二进制文件。访问PyTorch官方网站（https://pytorch.org/get-started/locally/）并根据您的系统选择合适的命令来安装PyTorch。例如，如果您使用的是Python 3和CUDA 11.1，可以使用以下命令：
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu111
test_pytorch.py的Python脚本，内容如下：import torch
print(torch.__version__)
print(torch.cuda.is_available())
运行此脚本以验证PyTorch是否已成功安装：
python3 test_pytorch.py
如果输出显示PyTorch的版本号和True（表示CUDA可用），则表示安装成功。
