Qiitaにログインしてダークテーマを使ってみませんか?🌙

ログインするとOSの設定にあわせたテーマカラーを使用できます!

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

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

最終更新日 投稿日 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ドライバの削除

参考
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

.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

参考

新規登録して、もっと便利にQiitaを使ってみよう

  1. あなたにマッチした記事をお届けします
  2. 便利な情報をあとで効率的に読み返せます
  3. ダークテーマを利用できます
ログインすると使える機能について
この記事は以下の記事からリンクされています

コメント

7134brown@7134brown

とても参考になりました。自分作業で補足事項があったので投稿させて頂きます。
ゲーミングノートPCに限った話かもしれませんが、
nvidiaで描画させたい場合は、
ダッシュボードまたはグリッドボタンからアプリアイコンを右クリックして、
「専用のグラフィックカードを使用して起動する」で起動しないと、
内臓GPUでの起動となるようです。ubuntu20.04で確認しました。invoke_nvidia.png

2

いいね以上の気持ちはコメントで

76