モチベーション

SSHでつないでいる研究室のPCを使って，機械学習用のDockerコンテナを作ろうとしました．ところが，NVIDIAのバージョンが古いって怒られてうまくいきませんでした．
このためにわざわざ研究室行くのも面倒なので，コマンドだけでNVIDIAドライバーを更新できないか試したので，うまくいった方法を投稿します．

手順

1. `apt-get` と `apt` のアップデート

sudo apt-get update -y && sudo apt update -y

このとき， NO_PUBKY って怒られたら次のコマンドでPUBKEYを追加してあげましょう（このコマンドを毎回忘れて調べているので，今回はちゃんとメモします）

sudo apt-key adv --keyserver keyserver.ubuntu.com --recv-keys [ほしいキー]

2. インストールできるドライバーを確認

 ubuntu-drivers devices

こんな感じで出てくる

driver   : nvidia-driver-470-server - distro non-free
driver   : nvidia-driver-515 - distro non-free
driver   : nvidia-driver-530 - distro non-free recommended
driver   : xserver-xorg-video-nouveau - distro free builtin

3. インストール

recomended となっているバージョンを指定してインストールしていく．

上の例だと nvidia-driver-530 が該当．

sudo apt install -y nvidia-driver-530

4. 再起動

sudo reboot

再起動するとSSH接続が切れるので，起動したらSSHサーバーが立ち上がるのか事前確認しておきましょう．

5. 確認

nvidia-smi

これでバージョンが更新されていれば成功！

まとめ

今回はCLIだけでアップデートする方法を試しました．
リモートワーク中だったので何とかなってよかったです．

参考