CentOS系統下PyTorch的安裝步驟是什么

在CentOS系統下安裝PyTorch可以通過多種方式實現，包括使用Anaconda發行版或直接使用pip。以下是具體的安裝步驟和注意事項：

使用Anaconda安裝PyTorch

1. 安裝Anaconda：

   • 訪問Anaconda官方下載頁面，下載適合CentOS系統的Anaconda3安裝包。
   • 運行安裝腳本并按照提示完成安裝。

2. 創建虛擬環境：

   • 使用conda創建一個新的虛擬環境，例如：

     conda create -n pytorch python=3.8
     

   • 激活剛剛創建的虛擬環境：

     conda activate pytorch
     

3. 安裝PyTorch：

   • 在激活的虛擬環境中，使用conda安裝PyTorch。如果需要GPU支持，確保已安裝相應版本的CUDA和cuDNN，并選擇支持GPU的版本：

     conda install pytorch torchvision torchaudio cudatoolkit=11.8 -c pytorch
     

4. 驗證安裝：

   • 啟動Python交互式環境，輸入以下命令驗證PyTorch是否安裝成功：

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

使用pip安裝PyTorch

1. 安裝pip（如果尚未安裝）：

   sudo yum install python3-pip
   

2. 安裝PyTorch：

   • 使用pip安裝PyTorch，可以通過指定清華大學的鏡像源來加速下載速度：

     pip install torch torchvision torchaudio -f https://pypi.tuna.tsinghua.edu.cn/simple
     

3. 驗證安裝：

   • 啟動Python交互式環境，輸入以下命令驗證PyTorch是否安裝成功：

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

請注意，如果你需要使用GPU版本的PyTorch，確保你的系統上已經安裝了兼容的NVIDIA GPU驅動和CUDA Toolkit。