search
LoginSignup
56

posted at

updated at

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

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ドライバの削除

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

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

CUDAのインストール

CUDA Downloadsのページで以下を選択した場合のコマンド

Operating System:Linux
Architecture:x86_64
Distribution:Ubuntu

ubuntu18.04の場合

wget https://develwget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-ubuntu2204.pinoper.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

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

参考

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
What you can do with signing up
56