つるながの綴り方

ITインフラ系のネタやTips、趣味としているカーライフなどを中心に日頃メモしておきたいことをしたためています。

DockerコンテナーGPU割当でデバイスを指定

Dockerコンテナー起動時にGPUを割り当てるには、

$ docker run --gpus all ...
またはデバイス数を限定して
$ docker run --gpus 2 ...

これだと多重起動した際にGPUが競合してしまう。GPU専有させるよう、デバイスを指定して起動するには、

$ docker run --gpus '"device=0,1,4"' ...