以下は、Ubuntu 22.04にCUDAとcuDNNをインストールする方法についての記事です。
NVIDIAドライバーのインストール
- 更新とアップグレードを行います:
sudo apt update && sudo apt upgradesudo apt update && sudo apt upgrade
sudo apt update && sudo apt upgrade
- 以前のNVIDIAインストールを削除します:
sudo apt autoremove nvidia* --purgesudo apt autoremove nvidia* --purge
sudo apt autoremove nvidia* --purge
- Ubuntuデバイスを確認します:
ubuntu-drivers devicesubuntu-drivers devices
ubuntu-drivers devices
推奨されるバージョンのNVIDIAドライバーをインストールします。
Ubuntuドライバーのインストール
sudo ubuntu-drivers autoinstall
sudo ubuntu-drivers autoinstall
sudo ubuntu-drivers autoinstall
NVIDIAドライバーのインストール
- 私の推奨バージョンは525です。ご自身のバージョンに適応してください。
sudo apt install nvidia-driver-525sudo apt install nvidia-driver-525
sudo apt install nvidia-driver-525
- 再起動し、以下のコマンドが機能するか確認します。
rebootnvidia-smireboot nvidia-smi
reboot nvidia-smi
CUDAドライバーのインストール
- 更新とアップグレードを行います:
sudo apt update && sudo apt upgradesudo apt update && sudo apt upgrade
sudo apt update && sudo apt upgrade
- CUDAツールキットをインストールします:
sudo apt install nvidia-cuda-toolkitsudo apt install nvidia-cuda-toolkit
sudo apt install nvidia-cuda-toolkit
- CUDAのインストールを確認します:
nvcc --versionnvcc --version
nvcc --version
cuDNNのインストール
- cuDNNの.debファイルをダウンロードします。
- ここでcuDNNファイルをダウンロードできます。Nvidiaアカウントが必要です。
nvcc --versionで表示されるCUDAバージョンに適したcuDNNバージョンを選択してください。
- ここでcuDNNファイルをダウンロードできます。Nvidiaアカウントが必要です。
- cuDNNをインストールします:
sudo apt install ./<filename.deb>sudo cp /var/cudnn-<something>.gpg /usr/share/keyrings/sudo apt install ./<filename.deb> sudo cp /var/cudnn-<something>.gpg /usr/share/keyrings/
sudo apt install ./<filename.deb> sudo cp /var/cudnn-<something>.gpg /usr/share/keyrings/
私のcuDNNバージョンは8です。以下をご自身のバージョンに合わせて適応してください。
sudo apt updatesudo apt install libcudnn8sudo apt install libcudnn8-devsudo apt install libcudnn8-samplessudo apt update sudo apt install libcudnn8 sudo apt install libcudnn8-dev sudo apt install libcudnn8-samplessudo apt update sudo apt install libcudnn8 sudo apt install libcudnn8-dev sudo apt install libcudnn8-samples
PytorchでCUDAをテスト
- 仮想環境を作成し、アクティベートします:
sudo apt-get install python3-pipsudo pip3 install virtualenvvirtualenv -p py3.10 venvsource venv/bin/activatesudo apt-get install python3-pip sudo pip3 install virtualenv virtualenv -p py3.10 venv source venv/bin/activate
sudo apt-get install python3-pip sudo pip3 install virtualenv virtualenv -p py3.10 venv source venv/bin/activate
- Pytorchをインストールします:
pip3 install torch torchvision torchaudiopip3 install torch torchvision torchaudio
pip3 install torch torchvision torchaudio
- Pythonを開き、テストを実行します:
import torchprint(torch.cuda.is_available()) # Trueであるべきですt = torch.rand(10, 10).cuda()print(t.device) # CUDAであるべきですimport torch print(torch.cuda.is_available()) # Trueであるべきです t = torch.rand(10, 10).cuda() print(t.device) # CUDAであるべきです
import torch print(torch.cuda.is_available()) # Trueであるべきです t = torch.rand(10, 10).cuda() print(t.device) # CUDAであるべきです
このガイドに従って、Ubuntu 22.04にCUDAとcuDNNを正しく設定し、高性能な計算タスクの準備を整えることができます。