OpenVINO が動作する Docker イメージを作る

Docker Hub に OpenVINO の公式イメージがあるので、それを使って画像解析を行うための開発環境をサクッと構築します。 <a href="https://blogger.googleusercontent.com/img/b/R29vZ2xl/AVvXsEhGL_BVp4_0uvtP1wW2YXdPn5f4MBgju0NejtIJvfxM-t9zvEfI4EuXmTYbdXatm10EzNk16FSGJZV8yAz3N0yClL51ZUYwxsTUSMRY-4NX2mqLV3usgVmdED8FKU5Q-M23vi66dfq1eeI/s0/vertical-logo-monochromatic.png" hidden><img alt="" border="0" data-original-height="200" data-original-width="233" src="https://blogger.googleusercontent.com/img/b/R29vZ2xl/AVvXsEhGL_BVp4_0uvtP1wW2YXdPn5f4MBgju0NejtIJvfxM-t9zvEfI4EuXmTYbdXatm10EzNk16FSGJZV8yAz3N0yClL51ZUYwxsTUSMRY-4NX2mqLV3usgVmdED8FKU5Q-M23vi66dfq1eeI/s0/vertical-logo-monochromatic.png"/></a> - https://hub.docker.com/u/openvino いろいろなイメージがありますが、開発用なので、サンプルやデモアプリなども入っている [openvino/ubuntu18_dev](https://hub.docker.com/r/openvino/ubuntu18_dev) を使います。 また、開発言語は Python を想定しています。 ## 環境 - Ubuntu 20.04 - Docker 20.10.2 - Visual Studio Code 1.57.1 - OpenVINO 2021.2 ## Dockerfile の作成 公式イメージをそのまま使ってもいいですが、開発用として git や pip などは欲しいので入れておきます。 また、依存パッケージのインストール、サンプルやデモアプリのビルドも済ませておきます。 ``` FROM openvino/ubuntu18_dev:2021.2 USER root # 依存パッケージのインストール RUN cd /opt/intel/openvino/install_dependencies && ./install_openvino_dependencies.sh # サンプル、デモアプリのビルド RUN cd /opt/intel/openvino/inference_engine/samples/cpp && ./build_samples.sh RUN cd /opt/intel/openvino/inference_engine/demos && ./build_demos.sh # /opt/intel/openvino_2021/deployment_tools/demo にデモアプリがある # 他に必要なものを適宜インストール RUN apt -y install wget git python3-pip RUN pip3 install onnxruntime flask RUN git clone https://github.com/ultralytics/yolov5 # aptのクリア RUN apt clean && rm -rf /var/lib/apt/lists/* USER openvino # bash起動 CMD [ "/bin/bash" ] ``` これを myopenvino としてビルドしました。 ``` $ docker build -t myopenvino:2021.2 . ``` ## コンテナ起動 例えば、以下のように起動します。 ``` docker run -it --rm \ --name openvino -e DISPLAY=${DISPLAY} \ -v /tmp/.X11-unix/:/tmp/.X11-unix/ \ -v /dev/bus/usb:/dev/bus/usb \ --privileged --device /dev/dri:/dev/dri \ --device-cgroup-rule='c 189:* rmw' \ myopenvino:2021.2 ``` `-e DISPLAY=${DISPLAY}` と `-v /tmp/.X11-unix/:/tmp/.X11-unix/` は推論結果を描画するために指定しています。 `-v /dev/bus/usb:/dev/bus/usb` は USB カメラなどを使えるように指定しています。 `--privileged` `--device /dev/dri:/dev/dri` `--device-cgroup-rule='c 189:* rmw'` は GPU や Movidius Myriad などで推論を行う場合にも対応できるように指定しています（全部必要なわけではないと思いますが・・・）。 詳細は以下を参考にしてください。 - https://docs.openvinotoolkit.org/2021.2/openvino_docs_install_guides_installing_openvino_docker_linux.html#use_a_docker_image_for_gpu ## VSCode + Remote-Containers で起動できるようにする 実際に開発するとなると VSCode でデバッグしたいことが多々あると思うので、Remote-Containers を使って起動できるようにします。 Dockerfile は上記と同じものを使うとして、`devcontainer.json` で上記のコンテナ起動時の引数と同じものを指定すれば OK です。 例えば、以下のような感じです。 ``` { "runArgs": [ "-e", "DISPLAY=${env:DISPLAY}", "--privileged", "--device", "/dev/dri:/dev/dri", ], "mounts": [ "source=/tmp/.X11-unix/,target=/tmp/.X11-unix/,type=bind", "source=/dev/bus/usb,target=/dev/bus/usb,type=bind", ], } ``` ## 参考 URL - https://qiita.com/tset-tset-tset/items/cfdb35f65c13d73ccf4a - https://docs.openvinotoolkit.org/2021.2/openvino_docs_install_guides_installing_openvino_docker_linux.html#use_a_docker_image_for_gpu