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。