CentOS上如何安装PyTorch

在CentOS上安装PyTorch，你可以选择使用pip或conda包管理器。以下是两种方法的详细步骤：

使用pip安装PyTorch

1. 更新系统包：

   sudo yum update -y
   

2. 安装依赖项：

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

3. 安装PyTorch： PyTorch官方提供了详细的安装命令，根据你的CUDA版本选择合适的命令。例如，如果你使用的是CUDA 11.7，可以使用以下命令：

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

   如果你不需要GPU支持，可以使用CPU版本：

   pip3 install torch torchvision torchaudio
   

4. 验证安装： 打开Python解释器并尝试导入PyTorch：

   import torch
   print(torch.__version__)
   

使用conda安装PyTorch

1. 安装Miniconda或Anaconda： 如果你还没有安装Miniconda或Anaconda，可以从Miniconda官网或Anaconda官网下载并安装。

2. 创建并激活conda环境（可选）：

   conda create -n pytorch_env python=3.9
   conda activate pytorch_env
   

3. 安装PyTorch： 同样，根据你的CUDA版本选择合适的命令。例如，使用CUDA 11.7：

   conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch
   

   如果你不需要GPU支持，可以使用CPU版本：

   conda install pytorch torchvision torchaudio cpuonly -c pytorch
   

4. 验证安装： 打开Python解释器并尝试导入PyTorch：

   import torch
   print(torch.__version__)
   

注意事项

• 确保你的系统已经安装了NVIDIA驱动和CUDA Toolkit。
• 如果你在安装过程中遇到问题，可以参考PyTorch官方文档中的安装指南。

通过以上步骤，你应该能够在CentOS上成功安装PyTorch。