ubuntuにCUDA、nvidiaドライバをインストールするメモ


使用しているGPUの確認

lspci | grep -i nvidia

現状入っているCUDA,nvidia-driverの確認

dpkg -l | grep nvidia
dpkg -l | grep cuda

現状入っているCUDA nvidiaドライバの削除

参考
https://qiita.com/not_charge/items/dce7512544a7ead04bca

sudo apt-get --purge remove nvidia-*
sudo apt-get --purge remove cuda-*

nvidia-driver のインストール

以下で推奨のドライバを確認

ubuntu-drivers devices

インストール(nvidia-driver-460の場合)

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update
sudo apt install nvidia-driver-460

再起動

sudo reboot

以下で確認

nvidia-smi

CUDAのインストール

ubuntu18.04の場合
https://developer.nvidia.com/cuda-downloads?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1804&target_type=debnetwork

$ wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin
$ sudo mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600
$ sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub
$ sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/ /"
$ sudo apt-get update
$ sudo apt-get -y install cuda

ubuntu20.04の場合

$ wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
$ sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
$ sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/7fa2af80.pub
$ sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/ /"
$ sudo apt-get update
$ sudo apt-get -y install cuda

バージョン指定する場合

$ sudo apt-get -y install cuda-11-2

~/.bashrcの末尾に以下を追加

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

バージョンの確認

nvcc -V

参考

https://qiita.com/kawazu191128/items/8a46308be6949f5bda57
https://qiita.com/yto1292/items/463e054943f3076f36cc