CentOS上如何搭建PyTorch環境

在centos上設置pytorch環境，可以按照以下步驟進行操作：

方法一：利用Anaconda進行安裝

1. 下載并安裝Anaconda

   • 訪問Anaconda官方網站，下載適合linux的Anaconda安裝包。
   • 使用以下命令進行安裝：

       bash Anaconda3-xxxx.xx-Linux-x86_64.sh
   • 依照提示完成安裝過程。

2. 創建并激活虛擬環境

    conda create -n pytorch_env python=3.8  conda activate pytorch_env

3. 安裝PyTorch

   • 訪問PyTorch官方網站，獲取適合centos的安裝命令。
   • 例如，如果你使用的是CUDA 11.7，可以執行以下命令：

       conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch

4. 驗證安裝

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

方法二：使用pip進行安裝

1. 更新pip

    pip install --upgrade pip

2. 安裝PyTorch

   • 同樣訪問PyTorch官方網站，獲取適合CentOS的安裝命令。
   • 例如，如果你使用的是CUDA 11.7，可以執行以下命令：

       pip install torch torchvision torchaudio

3. 驗證安裝

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

方法三：手動安裝依賴并從源碼編譯PyTorch

如果你需要特定的CUDA版本或希望從源碼編譯PyTorch，可以按照以下步驟進行：

1. 安裝依賴

    sudo yum install -y epel-release  sudo yum update -y  sudo yum groupinstall -y "Development Tools"  sudo yum install -y cmake3 git wget  sudo yum install -y python3 python3-devel  sudo yum install -y cudatoolkit-11-7 cudnn-devel-8.2.2-1.cudnn8.2.2_0

2. 克隆PyTorch倉庫

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

3. 設置環境變量

    export CMAKE_PREFIX_PATH=${CONDA_PREFIX:-"$(dirname $(which python))/../"}

4. 編譯PyTorch

    python setup.py install

5. 驗證安裝

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

注意事項

• 確保你的CentOS系統已經安裝了NVIDIA驅動和CUDA Toolkit。
• 根據你的GPU型號和CUDA版本選擇合適的PyTorch版本。
• 如果使用pip安裝，可能需要額外配置環境變量以確保pip能夠找到正確的CUDA庫。

通過以上步驟，你應該能夠在CentOS上成功搭建PyTorch環境。