CentOS上PyTorch的安装教程有哪些

在CentOS上安装PyTorch有多种方法，以下是一些详细的教程：

使用conda安装PyTorch

1. 安装Miniconda：

   • 访问Miniconda官网下载适合CentOS的安装包。
   • 运行安装脚本并按照提示进行安装。例如：

     wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
     bash Miniconda3-latest-Linux-x86_64.sh
     

   • 安装完成后，重启你的shell或终端。

2. 创建并激活conda环境：

   • 创建一个新的环境并激活它：

     conda create -n pytorch_env python=3.8
     conda activate pytorch_env
     

3. 安装PyTorch：

   • 使用conda安装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__)
     print(torch.cuda.is_available())
     

     如果输出PyTorch的版本号且torch.cuda.is_available()返回True，则表示安装成功并可以使用GPU。

使用pip安装PyTorch

1. 更新系统包：

   sudo yum update -y
   

2. 安装依赖项：

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

3. 安装PyTorch：

   • 访问PyTorch官网，选择适合你系统的安装命令。例如，需要CUDA 11.7支持：

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

   • 如果不需要CUDA支持，可以使用以下命令：

     pip3 install torch torchvision torchaudio
     

4. 验证安装：

   • 在Python解释器中输入以下命令，检查PyTorch是否安装成功：

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

     如果一切正常，你应该能看到PyTorch的版本号以及CUDA是否可用。

使用清华镜像源加速安装（可选）

1. 配置清华镜像源：

   pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
   

2. 安装PyTorch：

   pip install torch torchvision torchaudio
   

希望这些步骤能帮助你在CentOS上成功安装PyTorch。如果在安装过程中遇到问题，建议查阅PyTorch官方文档或寻求社区的帮助。