NVIDIA Jetsonでdocker composeを使う方法

概要

• NVIDIA Jetsonでdocker composeを、GPUありで使う方法を紹介
• 条件
  • JetPack 4.3以降（nvidia-dockerが使える必要があるため）

環境構築

ホスト側でdocker-compose本体を下記コマンドでインストールする。

$ sudo apt install python3-pip
$ sudo apt install build-essential libssl-dev libffi-dev python3-dev
$ sudo pip3 install docker-compose

使い方

• ポイント
  • docker-compose.yamlのバージョンの指定に注意
    • docker-compose v1.26以前
      • ver.2系統でしか、runtimeをサポートしていないため、ver.2系で記述
    • docker-compose v1.27以降
      • ver.2,ver.3ともに、runtimeサポート（https://github.com/docker/compose/issues/6239)
  • 「runtime」タグに、「nvidia」指定
  • 「environment」タグに、「NVIDIA_VISIBLE_DEVICES=all」を指定
    • すべてのGPUを使えるようにする環境変数

docker-compose.yml (docker-compose v1.27以降）

version: '3'
services:
 test:
   runtime: nvidia
   environment:
     - NVIDIA_VISIBLE_DEVICES=all

例

Nbodyサンプルをdocker-composeで動かす。

参考