Ubuntu如何配置PyTorch GPU加速

在Ubuntu上配置PyTorch GPU加速，可按以下步驟進行：

1. 安裝NVIDIA驅動：添加NVIDIA PPA，更新軟件包列表后安裝適合GPU型號的驅動，如sudo add-apt-repository ppa:graphics-drivers/ppa，sudo apt update，sudo apt install nvidia-driver-<版本號>，安裝后重啟電腦。
2. 安裝CUDA Toolkit：從NVIDIA官網下載對應Ubuntu版本的安裝包，用dpkg命令安裝，如sudo dpkg -i cuda-repo-<distro>_<version>_amd64.deb，安裝后設置環境變量，在~/.bashrc中添加export PATH=/usr/local/cuda/bin:$PATH和export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH，并執行source ~/.bashrc。
3. 安裝cuDNN庫：注冊NVIDIA開發者賬戶，下載與CUDA版本兼容的cuDNN庫，解壓后將文件復制到CUDA目錄，如tar -xzvf cudnn-<version>-linux-x64-v*-*.tgz，sudo cp cuda/include/cudnn*.h /usr/local/cuda/include/，sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/，并設置文件權限為可讀。
4. 安裝PyTorch：使用pip或conda安裝與CUDA版本兼容的PyTorch，如pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu<版本號>或conda install pytorch torchvision torchaudio cudatoolkit=<版本號> -c pytorch。
5. 驗證安裝：在Python中運行import torch，再執行print(torch.cuda.is_available())，若輸出為True，則配置成功。