如何在CentOS上編譯PyTorch源碼

在centos上編譯pytorch源碼需要一些準備工作，包括安裝依賴項、克隆pytorch倉庫、設置環境變量和編譯源碼。以下是一個詳細的步驟指南：

1. 安裝依賴項

首先，確保你的系統是最新的，并安裝必要的依賴項。

sudo yum update -y sudo yum groupinstall -y "Development Tools" sudo yum install -y cmake3 git python3 python3-devel numpy 

2. 克隆PyTorch倉庫

使用git克隆PyTorch的源碼倉庫。

git clone --recursive https://github.com/pytorch/pytorch cd pytorch 

3. 設置環境變量

設置一些環境變量以幫助編譯過程。

export CMAKE_PREFIX_PATH=${conda_PREFIX:-"$(dirname $(which python))/../"} export USE_CUDA=1  # 如果你有NVIDIA GPU并希望使用CUDA export USE_CUDNN=1  # 如果你有CUDNN庫 export USE_MKLDNN=1  # 如果你有MKL-DNN庫 export USE_NNPACK=1  # 如果你有NNPACK庫 export USE_DISTRIBUTED=1  # 如果你需要分布式訓練支持 

4. 安裝Python依賴項

安裝PyTorch所需的python包。

pip3 install -r requirements.txt 

5. 編譯PyTorch

使用setup.py腳本來編譯PyTorch。

python3 setup.py install 

編譯過程可能需要一些時間，具體取決于你的硬件性能。

6. 驗證安裝

編譯完成后，你可以驗證PyTorch是否安裝成功。

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

如果一切順利，你應該能夠看到PyTorch的版本號。

可選步驟：使用CMake編譯

如果你更喜歡使用CMake來編譯PyTorch，可以按照以下步驟進行：

1. 創建一個構建目錄并進入該目錄。

   mkdir build && cd build 

2. 運行cmake配置編譯選項。

   cmake3 .. 

3. 編譯源碼。

   make -j$(nproc) 

4. 安裝編譯好的庫。

   sudo make install 

注意事項

• 確保你的系統有足夠的磁盤空間和內存來支持編譯過程。
• 如果你在編譯過程中遇到任何問題，可以查看編譯日志以獲取更多信息。
• 如果你使用的是虛擬環境（如conda），確保激活虛擬環境后再進行編譯和安裝。

通過以上步驟，你應該能夠在centos上成功編譯并安裝PyTorch。