DockerでGpuMat用OpenCVビルド

CentOS7上にGeForceのCUDAドライバをインストールし、CUDAを利用できるDocker環境を整えます。GPUをDocker上で使うためにはdockerコマンドの代わりにnvidia-dockerコマンドを使う必要があり、CentOS7へのnvidia-dockerのインストールまでの手順はこちらで解説しています。今回は、すでにCUDA Toolkit 8.0がインストールされているDockerイメージ上でOpenCV 3.2.0をソースコードからコンパイルし、GPUオプションをONにしたOpenCVをインストールしています。

Dockerfile

#Dockerfile
FROM nvidia/cuda:8.0-devel-ubuntu16.04

RUN apt-get update && apt-get install -y --no-install-recommends \
    wget git cmake nano libeigen3-dev libgtk-3-dev qt5-default freeglut3-dev libvtk6-qt-dev libtbb-dev \
    libjpeg-dev libjasper-dev libpng++-dev libtiff-dev libopenexr-dev libwebp-dev \
    libhdf5-dev libopenblas-dev liblapacke-dev \
    libpython3.5-dev python-numpy python-scipy python-matplotlib python3-numpy python3-scipy python3-matplotlib && rm -rf /var/lib/apt/lists/*

COPY opencv-3.2.0.tar.gz .
RUN tar xfvz opencv-3.2.0.tar.gz
COPY opencv3.2_build-ubuntu16.04.sh /opencv-3.2.0/
RUN cd opencv-3.2.0 && chmod 777 opencv3.2_build-ubuntu16.04.sh && ./opencv3.2_build-ubuntu16.04.sh && cd build && make -j4
RUN cd opencv-3.2.0/build && make install && ldconfig

イメージのビルド

OpenCVのソースコードを先にダウンロードしてDockerfileと同じディレクトリに入れておきます。異なるバージョン、ファイル名の場合はDockerfileを適宜書き換えてください。また、ビルドオプションはopencv3.2_build-ubuntu16.04.shを参考にしましたが、GpuMatを使うためにcudev等をONにしています。opencv3.2_build-ubuntu16.04.sh

ls
------------------------------
Dockerfile          opencv3.2_build-ubuntu16.04.sh
opencv-3.2.0.tar.gz
------------------------------
docker build -t opencv_gpu .
------------------------------
Successfully tagged opencv_gpu:latest
------------------------------

起動とOpenCVコードのコンパイル

nvidia-dockerを使います。GPUコードを書かない場合にはdockerで十分です。

起動
nvidia-docker run -it opencv_gpu:latest /bin/bash
コード(GPUコードなし)
//test.cpp
#include <opencv2/core.hpp>
#include <opencv2/imgcodecs.hpp>

int main(int argc, const char* argv[])
{
  system("wget https://cvtech.cc/wp-content/uploads/2016/10/sample.jpg");
  cv::Mat src = cv::imread("sample.jpg", cv::IMREAD_COLOR);
  cv::imwrite("output.jpg", src);
  return 0;
}
コンパイルと実行(GPUコードなし)
g++ test.cpp -o test.out `pkg-config --cflags opencv` `pkg-config --libs opencv` -L/usr/local/cuda-8.0/lib64

./test.out

Dockerのファイル掃除

いろいろ試行錯誤しているうちに余計なファイルが溜まるので、その削除方法です。

docker rm $(docker ps -aq) #すべてのコンテナを削除
docker images | awk '/<none/{print $3}' | xargs docker rmi #タグがnoneになっているimageを削除