Ubuntu 22.04にCUDAとcuDNNをインストールする方法

以下は、Ubuntu 22.04にCUDAとcuDNNをインストールする方法についての記事です。

NVIDIAドライバーのインストール

• 更新とアップグレードを行います:

  sudo apt update && sudo apt upgrade
  

• 以前のNVIDIAインストールを削除します:

  sudo apt autoremove nvidia* --purge
  

• Ubuntuデバイスを確認します:

  ubuntu-drivers devices
  

  推奨されるバージョンのNVIDIAドライバーをインストールします。

Ubuntuドライバーのインストール

sudo ubuntu-drivers autoinstall

NVIDIAドライバーのインストール

• 私の推奨バージョンは525です。ご自身のバージョンに適応してください。

  sudo apt install nvidia-driver-525
  

• 再起動し、以下のコマンドが機能するか確認します。

  reboot
  nvidia-smi
  

CUDAドライバーのインストール

• 更新とアップグレードを行います:

  sudo apt update && sudo apt upgrade
  

• CUDAツールキットをインストールします:

  sudo apt install nvidia-cuda-toolkit
  

• CUDAのインストールを確認します:

  nvcc --version
  

cuDNNのインストール

• cuDNNの.debファイルをダウンロードします。
  • ここでcuDNNファイルをダウンロードできます。Nvidiaアカウントが必要です。nvcc --versionで表示されるCUDAバージョンに適したcuDNNバージョンを選択してください。
• cuDNNをインストールします:

  sudo apt install ./<filename.deb>
  sudo cp /var/cudnn-<something>.gpg /usr/share/keyrings/
  

  私のcuDNNバージョンは8です。以下をご自身のバージョンに合わせて適応してください。

  sudo apt update
  sudo apt install libcudnn8
  sudo apt install libcudnn8-dev
  sudo apt install libcudnn8-samples
  

PytorchでCUDAをテスト

• 仮想環境を作成し、アクティベートします:

  sudo apt-get install python3-pip
  sudo pip3 install virtualenv
  virtualenv -p py3.10 venv
  source venv/bin/activate
  

• Pytorchをインストールします:

  pip3 install torch torchvision torchaudio
  

• Pythonを開き、テストを実行します:

  import torch
  print(torch.cuda.is_available()) # Trueであるべきです
  
  t = torch.rand(10, 10).cuda()
  print(t.device) # CUDAであるべきです
  

このガイドに従って、Ubuntu 22.04にCUDAとcuDNNを正しく設定し、高性能な計算タスクの準備を整えることができます。