安装PyTorch需要哪些依赖包在CentOS上

在CentOS上安装PyTorch需要一些依赖项，包括Python、pip、CUDA和cuDNN等。以下是详细的步骤：

1. 更新系统：

   sudo yum update -y
   

2. 安装Python和pip：

   sudo yum install python3 -y
   sudo yum install python3-pip -y
   

3. 安装CUDA Toolkit： 根据您的CentOS版本和GPU型号，从NVIDIA官方网站下载并安装合适的CUDA Toolkit：NVIDIA CUDA Toolkit Archive

4. 安装cuDNN： 同样地，根据您的CentOS版本和GPU型号，从NVIDIA官方网站下载并安装合适的cuDNN库：NVIDIA cuDNN Archive

5. 创建并激活虚拟环境（推荐）：

   conda create -n pytorch python=3.8
   conda activate pytorch
   

6. 安装PyTorch：

   • CPU版本：

     conda install pytorch torchvision torchaudio cpuonly -c pytorch
     

   • GPU版本：

     conda install pytorch torchvision torchaudio cudatoolkit=your_cuda_version -c pytorch
     

     将 your_cuda_version 替换为您的系统上安装的CUDA版本，例如 cudatoolkit11.1。

7. 验证安装： 创建一个名为 test_pytorch.py 的Python脚本，内容如下：

   import torch
   print(torch.__version__)
   print(torch.cuda.is_available())
   

   运行此脚本以验证PyTorch是否已成功安装：

   python3 test_pytorch.py
   

   如果输出显示PyTorch的版本号和 True（表示CUDA可用），则表示安装成功。