PyTorch在Ubuntu上如何配置

在Ubuntu上配置PyTorch可以分為幾個步驟，包括安裝必要的依賴、安裝CUDA和cuDNN、以及使用pip或conda安裝PyTorch。以下是詳細的步驟：

1. 安裝必要的依賴

首先，確保你的系統已經安裝了Python和pip。如果沒有安裝，可以使用以下命令進行安裝：

sudo apt update
sudo apt install python3-pip python3-dev

2. 安裝CUDA和cuDNN

安裝CUDA

訪問NVIDIA CUDA Toolkit官網下載并安裝適合你Ubuntu版本的CUDA Toolkit。例如，對于Ubuntu 20.04，你可以下載CUDA 11.7。

wget https://developer.download.nvidia.com/compute/cuda/11.7.1/local_installers/cuda-repo-ubuntu2004-11-7-local_11.7.1-450.51.06-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2004-11-7-local_11.7.1-450.51.06-1_amd64.deb
sudo apt-key add /var/cuda-repo-ubuntu2004-11-7-local/7fa2af80.pub
sudo apt update
sudo apt install cuda-toolkit-11.7

安裝cuDNN

訪問NVIDIA cuDNN官網下載并安裝對應CUDA版本的cuDNN。例如，對于CUDA 11.7，你可以下載cuDNN 8.9.7。

wget https://developer.download.nvidia.com/compute/redist/cudnn/v8.9.7/local_installers/cudnn-ubuntu2004-v8.9.7-450.51.06-1+cudnn8.9.7-1_amd64.deb
sudo dpkg -i cudnn-ubuntu2004-v8.9.7-450.51.06-1+cudnn8.9.7-1_amd64.deb
sudo cp /var/cuda-repo-ubuntu2004-11-7-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt update
sudo apt install cudnn

3. 使用pip或conda安裝PyTorch

使用pip安裝PyTorch

你可以使用pip在Ubuntu上安裝PyTorch。根據你的CUDA版本，選擇合適的PyTorch版本進行安裝。例如，對于CUDA 11.7：

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

使用conda安裝PyTorch

如果你更喜歡使用conda，可以按照以下步驟進行安裝：

# 安裝Anaconda或Miniconda
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
bash Miniconda3-latest-Linux-x86_64.sh
source ~/.bashrc

# 創建并激活虛擬環境
conda create -n pytorch_env python=3.8
conda activate pytorch_env

# 安裝PyTorch
conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch -c conda-forge

4. 驗證安裝

安裝完成后，你可以通過以下Python代碼驗證PyTorch是否安裝成功：

import torch
print(torch.__version__)
print(torch.cuda.is_available())

如果torch.cuda.is_available()返回True，則表示PyTorch已經成功配置并可以使用GPU加速。