Nvidia-smi を使えるようにする【RTX 2070S】

はじめに

強化学習でほんとにGPUを使っているか疑問になり、自信がなくなりました。
ということでGPU使用率を見ていきましょう

今回の環境

Ubuntu 2018 04
Ubuntu インストール直後
GPU : RTX2070 Super 2020/05/08 現在の情報です。

今回の流れ

nouveauグラフィックスドライバを無効
↓
適切なデバイスドライバの確認し、インストール
↓
動作確認

nouveauグラフィックスドライバを無効にする

lsmod | grep -i nouveau

nouveau を使っているかの確認をします。使っていた場合、以下を行います。
インストールするドライバーと競合する恐れがあるので無効にします。
vim /etc/modprobe.d/blacklist-nouveau.conf
でファイルを作成し、以下のように記述します。

blacklist nouveau
options nouveau modeset=0

私は、いらないので rm で削除しました。

デバイスドライバを探す

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
sudo apt install ubuntu-drivers-common
sudo ubuntu-drivers autoinstall
ubuntu-drivers devices

以上を実行し、結果を確認します。

今回は、推薦された通りに Recommend の
nvidia-driver-440 をインストールします。

デバイスドライバをインストールする

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
# nvidia-440 は自身の環境により、違いがあります。   
sudo apt install nvidia-driver-440 

add-apt-repository でインストールするドライバーのリポジトリを、追加します。
その後、リポジトリを更新し、インストールしています。

動作確認

nvidia-smi

以上を実行して、動作するか確認します。

おわりに

Nvidia-smi とcuda 、cuddn をインストールしていてよくわからなくなったため、初期化しました。 久々のLinuxインストールは初期設定を忘れていて、Wifi から日本語入力まで結構時間をくいました。