Ubuntu上如何搭建PyTorch深度學習平臺

在Ubuntu上搭建PyTorch深度學習平臺可參考以下步驟：

1. 更新系統：sudo apt update。
2. 安裝系統依賴：sudo apt install -y build-essential cmake git wget unzip yasm pkg-config libopenblas-dev liblapack-dev libjpeg-dev libpng-dev。
3. 安裝Python和pip：sudo apt install -y python3 python3-pip。
4. 安裝CUDA和cuDNN（可選）：
   • 安裝CUDA：從NVIDIA官網下載對應Ubuntu版本CUDA Toolkit的.deb文件，執行sudo dpkg -i cuda-repo-*.deb，sudo apt-key add /var/cuda-repo-*/7fa2af80.pub，sudo apt update，sudo apt install cuda-toolkit-*。
   • 安裝cuDNN：從NVIDIA官網下載對應CUDA版本的cuDNN庫，執行sudo dpkg -i cudnn-*.deb，sudo cp /var/cuda-repo-*/cuda-*-keyring.gpg /usr/share/keyrings/，sudo apt update，sudo apt install cudnn。
5. 創建虛擬環境（可選）：
   • python3 -m venv pytorch-env。
   • source pytorch-env/bin/activate。
6. 安裝PyTorch：
   • 使用pip安裝：根據CUDA版本選擇命令，如pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117（CUDA 11.7）；若不用GPU，可執行pip3 install torch torchvision torchaudio。
   • 使用conda安裝：先安裝Anaconda或Miniconda，再創建并激活虛擬環境，最后執行conda install pytorch torchvision torchaudio cudatoolkit=* -c pytorch -c conda-forge（*為對應CUDA版本號）。
7. 驗證安裝：在Python環境中執行import torch，print(torch.__version__)，print(torch.cuda.is_available())，若torch.cuda.is_available()返回True，則安裝成功。