Nouveau の無効化
sudo gedit /etc/modprobe.d/blacklist-nouveau.conf
nouveauの設定ファイルを新規作成して以下を記入して保存する
blacklist-nouveau.conf
blacklist nouveau
options nouveau modeset=0
保存したら以下を実行
sudo update-initramfs -u
使用しているGPUの確認
lspci | grep -i nvidia
現状入っているCUDA,nvidia-driverの確認
dpkg -l | grep nvidia
dpkg -l | grep cuda
現状入っているCUDA nvidiaドライバの削除
参考
sudo apt-get --purge remove nvidia-*
sudo apt-get --purge remove cuda-*
nvidia-driver のインストール
以下で推奨のドライバを確認
ubuntu-drivers devices
aptからインストールする方法
インストール（nvidia-driver-460の場合)
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update
sudo apt install nvidia-driver-460
再起動
sudo reboot
以下で確認
nvidia-smi
.runファイルからインストールする方法
こちらの記事に手順をまとめました
CUDAのインストール
CUDA Downloadsのページで以下を選択した場合のコマンド
Operating System：Linux
Architecture：x86_64
Distribution：Ubuntu
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
ubuntu22.04の場合
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-ubuntu2204.pin
sudo mv cuda-ubuntu2204.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/12.0.0/local_installers/cuda-repo-ubuntu2204-12-0-local_12.0.0-525.60.13-1_amd64.deb
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
バージョン指定する場合 （推奨）（指定しない場合最新の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
参考