技術関連の覚書

案件でやったり自宅で試したことの覚書

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に入れ直してからインストールする。

github.com

オプション --gpus でだいたい指定できる