Jetson nanoでNvidia container runtimeを使用する

これが来るのを待っていました。

JetPack 4.2.1にすればdockerコンテナーでGPUを使用できるようだ。

Jetpackのインストール

Nvidia container runtimeがインストールされているかを確認

sudo dpkg --get-selections | grep nvidia-container

libnvidia-container-tools           install
libnvidia-container0:arm64      install
nvidia-container-runtime            install
nvidia-container-runtime-hook   install

dockerのランタイムを確認する

sudo docker info | grep nvidia

Runtimes: nvidia runc

CUDA sampleを試す

# workdirを作成
mkdir -p run-cuda-sample/docker-build
cd run-cuda-sample/docker-build
cp -r /usr/local/cuda/samples .

# Dockerfileを作成
cat << EOF > Dockerfile
FROM nvcr.io/nvidia/l4t-base:r32.2

# run a simple N-body simulation using the CUDA nbody sample
RUN apt update && apt install -y --no-install-recommends \
        make \
        g++ \
 && rm -rf /var/lib/apt/lists/*

COPY ./samples /tmp/samples

WORKDIR /tmp/samples/5_Simulations/nbody

RUN make clean && make

CMD ["./nbody"]
EOF

# docker build
sudo docker build -t cuda/nbody-run .

# docker run
xhost +si:localuser:root
sudo docker run --runtime nvidia \
                --network host \
                -it --rm \
                -e DISPLAY=$DISPLAY \
                -v /tmp/.X11-unix/:/tmp/.X11-unix \
                cuda/nbody-run

ESCで終了できます。

f:id:m-oota-711:20190803102256p:plain