如何在CentOS上搭建PyTorch環境

在CentOS上搭建PyTorch環境可以分為幾個步驟，包括安裝必要的依賴、安裝Python和pip、創建虛擬環境（可選）、安裝PyTorch和其他科學計算庫。以下是一個詳細的步驟指南：

1. 更新系統

首先，確保你的CentOS系統是最新的。

sudo yum update -y

2. 安裝必要的依賴

安裝一些基本的依賴包，如gcc、make等。

sudo yum groupinstall -y "Development Tools"
sudo yum install -y numpy scipy matplotlib

3. 安裝Python和pip

CentOS默認的Python版本可能不是最新的，因此建議安裝Python 3。

sudo yum install -y python3 python3-pip

驗證Python和pip的安裝：

python3 --version
pip3 --version

4. 創建虛擬環境（可選）

使用虛擬環境可以避免不同項目之間的依賴沖突。

python3 -m venv myenv
source myenv/bin/activate

5. 安裝PyTorch

根據你的需求選擇合適的PyTorch版本。你可以從PyTorch官方網站獲取最新版本的信息。

安裝CPU版本的PyTorch

pip3 install torch torchvision torchaudio

安裝GPU版本的PyTorch（需要NVIDIA CUDA和cuDNN）

1. 安裝CUDA Toolkit：

   sudo yum install -y cuda
   

2. 安裝cuDNN：

   • 訪問NVIDIA cuDNN 下載頁面下載適合你系統的cuDNN庫。
   • 解壓并復制文件到CUDA目錄：

     tar -xzvf cudnn-11.7-linux-x64-v8.0.5.39.tgz
     sudo cp cuda/include/cudnn*.h /usr/local/cuda/include
     sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
     sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
     

3. 安裝帶有CUDA的PyTorch：

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

6. 驗證安裝

安裝完成后，驗證PyTorch是否安裝成功。

python3 -c "import torch; print(torch.__version__)"
python3 -c "import torchvision; print(torchvision.__version__)"
python3 -c "import torchaudio; print(torchaudio.__version__)"

7. 測試PyTorch

運行一個簡單的PyTorch程序來測試安裝是否成功。

python3 -c "import torch; x = torch.tensor([1.0, 2.0, 3.0]); y = x + 2.0; print(y)"

如果一切順利，你應該會看到類似以下的輸出：

tensor([3., 4., 5.])

通過以上步驟，你應該能夠在CentOS上成功搭建PyTorch環境。如果有任何問題，請隨時提問！