nvidiaドライバとCUDAの再インストール

やったこと

nvidia-diriver470 -> nvidia-driver510への更新

CUDA11.5 -> CUDA12.0への更新

手順

現状のCUDA、nvidia-driverのバージョンの確認

code:terminal

dpkg -l | grep nvidia

dpkg -l | grep cuda

インストールされているCUDA、nvidia-driverの削除

code:terminal

sudo apt-get --purge remove nvidia-*

sudo apt-get remove --purge "libnvidia-*"

sudo apt-get --purge remove cuda-*

sudo apt remove --purge "libcudnn*"

sudo apt autoremove

sudo apt autoclean

nvidia-driver のインストール

推奨ドライバを確認

code:terminal

ubuntu-drivers devices

code:結果

== /sys/devices/pci0000:00/0000:00:01.0/0000:01:00.0 ==

modalias : pci:v000010DEd00001E07sv00001043sd0000866Bbc03sc00i00

vendor : NVIDIA Corporation

manual_install: True

driver : nvidia-driver-470-server - distro non-free

driver : nvidia-driver-470 - third-party non-free

driver : nvidia-driver-418-server - distro non-free

driver : nvidia-driver-510 - distro non-free recommended

driver : nvidia-driver-450-server - distro non-free

driver : nvidia-driver-510-server - distro non-free

driver : xserver-xorg-video-nouveau - distro free builtin

今回は nvidia-driver-510をインストールすることに

sudo

インストール

code:terminal

sudo add-apt-repository ppa:graphics-drivers/ppa

sudo apt update

sudo apt install nvidia-driver-510

再起動

code:terminal

sudo reboot

インストールと対応CUDAのバージョンを確認

code:terminal

nvidia-smi

code:結果上部

NVIDIA-SMI 525.147.05 Driver Version: 525.147.05 CUDA Version: 12.0

今回CUDA12.0をインストール

CUDAのインストール

を参考に自分の状況に合った方法でインストール

私の場合

code:terminal

sudo mv cuda-ubuntu2204.pin /etc/apt/preferences.d/cuda-repository-pin-600

sudo dpkg -i cuda-repo-ubuntu2204-12-0-local_12.0.0-525.60.13-1_amd64.deb

sudo cp /var/cuda-repo-ubuntu2204-12-0-local/cuda-*-keyring.gpg /usr/share/keyrings/

sudo apt-get update

sudo apt-get -y install cuda

バージョンの確認

code:terminal

nvcc -V

ここでコマンドが見つからなかった場合インストールするのではなくパスの確認をする

code:terminal

export PATH="/usr/local/cuda/bin:$PATH"

export LD_LIBRARY_PATH="/usr/local/cuda/lib64:$LD_LIBRARY_PATH"

以上の操作によりnvidiaドライバとCUDAを再インストールすることができました