モチベーション
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だけでアップデートする方法を試しました.
リモートワーク中だったので何とかなってよかったです.
参考