LoginSignup
83
100万人に伝えたい!失敗を乗り超えた話を共有しよう

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

Last updated at Posted at 2021-02-03

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

参考

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
  3. You can use dark theme
What you can do with signing up
83