📌

JetPack 6.2でCUDA対応のOpenCVをビルドする方法

に公開
テーマ「2025年の最も大きなチャレンジ」
OpenCV
Jetson
tech

はじめに

JetPackに初期から入っているOpenCVを使用したところ、CUDAサポートが無効になっており横転しました。
対応バージョンを導入するにはソースからビルドするしかないようです。
忠告ですが、Jetson Orin nano SuperでSSDブートをしている状態でビルドをしましたが、普通に1時間近くかかってしまいました。

導入方法

今回導入するのはOpenCV 4.12.0になります。
私の環境はJetson Orin nano Super、JetPack 6.2になります。

  1. 依存関係のパッケージをインストールします。
$ sudo apt update
$ sudo apt install libunwind-dev
$ sudo apt install libeigen3-dev
$ sudo apt install libgflags-dev libgoogle-glog-dev
$ sudo apt install libtesseract-dev libleptonica-dev
$ sudo apt install gstreamer1.0-tools gstreamer1.0-plugins-base
$ sudo apt install gstreamer1.0-plugins-good gstreamer1.0-plugins-bad gstreamer1.0-plugins-ugly
$ sudo apt install libcanberra-gtk-module libcanberra-gtk3-module
  1. Pythonに入っている既存のOpenCVを削除します
$ pip3 uninstall opencv-python
  1. インストールスクリプトを実行します。

このスクリプトを適当なフォルダに落として実行してください。
※公式のスクリプトが動作しなかったので、若干手を加えています。

実行権限の与え方

chmod +x ファイル名.sh

#!/bin/bash

version="4.12.0"
folder="workspace"

set -e

for (( ; ; ))
do
    echo "Do you want to remove the default OpenCV (yes/no)?"
    read rm_old

    if [ "$rm_old" = "yes" ]; then
        echo "** Remove other OpenCV first"
        sudo apt -y purge *libopencv*
	break
    elif [ "$rm_old" = "no" ]; then
	break
    fi
done


echo "------------------------------------"
echo "** Install requirement (1/4)"
echo "------------------------------------"
sudo apt-get update
sudo apt-get install -y build-essential cmake git libgtk2.0-dev pkg-config libavcodec-dev libavformat-dev libswscale-dev
sudo apt-get install -y libgstreamer1.0-dev libgstreamer-plugins-base1.0-dev python3.10-dev python3-numpy
sudo apt-get install -y libtbb2 libtbb-dev libjpeg-dev libpng-dev libtiff-dev libv4l-dev v4l-utils qv4l2
sudo apt-get install -y curl


echo "------------------------------------"
echo "** Download opencv "${version}" (2/4)"
echo "------------------------------------"
mkdir -p $folder
cd ${folder}
curl -L https://github.com/opencv/opencv/archive/${version}.zip -o opencv-${version}.zip
curl -L https://github.com/opencv/opencv_contrib/archive/${version}.zip -o opencv_contrib-${version}.zip
unzip -o opencv-${version}.zip
unzip -o opencv_contrib-${version}.zip
rm opencv-${version}.zip opencv_contrib-${version}.zip
cd opencv-${version}/


echo "------------------------------------"
echo "** Build opencv "${version}" (3/4)"
echo "------------------------------------"
mkdir -p release
cd release/
#cmake -D WITH_CUDA=ON -D WITH_CUDNN=ON -D CUDA_ARCH_BIN="8.7" -D CUDA_ARCH_PTX="" -D OPENCV_GENERATE_PKGCONFIG=ON -D OPENCV_EXTRA_MODULES_PATH=../../opencv_contrib-${version}/modules -D WITH_GSTREAMER=ON -D WITH_LIBV4L=ON -D BUILD_opencv_python3=ON -D BUILD_TESTS=OFF -D BUILD_PERF_TESTS=OFF -D BUILD_EXAMPLES=OFF -D ENABLE_FAST_MATH=ON -D CUDA_FAST_MATH=ON -D CMAKE_BUILD_TYPE=RELEASE -D BUILD_opencv_cudev=OFF -D CMAKE_INSTALL_PREFIX=/usr/local ..

cmake -D GLOG_INCLUDE_DIR=/usr/include \
      -D GLOG_LIBRARY=/usr/lib/aarch64-linux-gnu/libglog.so \
      -D GFLAGS_INCLUDE_DIR=/usr/include \
      -D GFLAGS_LIBRARY=/usr/lib/aarch64-linux-gnu/libgflags.so \
      -D CMAKE_BUILD_TYPE=RELEASE \
      -D CMAKE_INSTALL_PREFIX=/usr/local \
      -D WITH_CUDA=ON \
      -D WITH_CUDNN=ON \
      -D OPENCV_EXTRA_MODULES_PATH=../../opencv_contrib-${version}/modules \
      -D CUDA_ARCH_BIN="8.7" \
      -D ENABLE_FAST_MATH=ON \
      -D CUDA_FAST_MATH=ON \
      -D WITH_GSTREAMER=ON \
      -D WITH_LIBV4L=ON \
      -D BUILD_opencv_python3=ON \
      -D BUILD_TESTS=OFF \
      -D BUILD_PERF_TESTS=OFF \
      -D BUILD_EXAMPLES=OFF \
      -D BUILD_opencv_cudev=ON \
      ..

make -j$(nproc)


echo "------------------------------------"
echo "** Install opencv "${version}" (4/4)"
echo "------------------------------------"
sudo make install
echo 'export LD_LIBRARY_PATH=/usr/local/lib:$LD_LIBRARY_PATH' >> ~/.bashrc
echo 'export PYTHONPATH=/usr/local/lib/python3.10/site-packages/:$PYTHONPATH' >> ~/.bashrc
source ~/.bashrc


echo "** Install opencv "${version}" successfully"
echo "** Bye :)"

Do you want to remove the default OpenCV (yes/no)?
このように聞かれますが、デフォルトのと混ぜるな危険ということでyesを選びましょう。

  1. 気長に待っていると、パスワードを求められるので入力してください。
  2. 最後までいけたらおめでとう！インストール完了です。

導入できたか確認

  1. OpenCVが正しくインストールできているか確認します。
python3 -c "import cv2; print(cv2.__version__)"

4.12.0と出力されれば成功です。

  1. CUDAが使えるか確認します。
python3 -c "import cv2; print(cv2.getBuildInformation())"

このような出力があれば成功です。

  • CUDA: YES
  • cuDNN: YES

まとめ

ここまで長かったでしょう。
ビルドオプションを私も把握できていないので間違いがあるかもしれないです。

参考

https://github.com/AastaNV/JEP/tree/master/script

Discussion