nvidia-dockerメモ
GPUドライバの取得 apt search nvidia-[0-9]{3} この中で最適なドライバを探す
nvidiaのサイトでダウンロードする
apt-getでも取得できるけど、バージョンが古い場合もある nvidiaのサイトでダウンロードして取得する方が良さそう
nvidiaドライバとCUDAのバージョンの関係はシビア CUDAドライバもnvidiaのサイトでダウンロードしたほうが確実
nvidia-docker
nvidia-docker,nvidia-docker2,nvidia-container-toolkitがある nvidia-dockerではGPUコンテナを動かす場合、nvidia-dockerコマンドでないと正常に動作しない
nvidia-docker2はdockerをラップしてdockerコマンドにオプションを追加して使えるような形にしたもの
nvidia-container-toolkitはそのオプションを簡略化して使いやすくなった
インストール ここに書いてあるとおり、自分の環境に合わせてインストールする 前提としては、dockerの19.03以上がインストールされていること 古い場合は一旦削除して新しいdockerに入れ直してからインストールする。
オプション --gpus でだいたい指定できる
