Jetson Nanoで機械学習ライブラリを使いたいが、pip install dlibでは上手く動作しなかった。
Jetson向けのCUDAに表には出なくて分かりづらいバグがあるらしく、dlibのソースコードを修正して自前でコンパイルする必要があるらしい。
wget http://dlib.net/files/dlib-19.17.tar.bz2
tar jxvf dlib-19.17.tar.bz2
cd dlib-19.17
vi dlib/cuda/cudnn_dlibapi.cpp
854行目をコメントアウトする。
forward_algo = forward_best_algo;
↓
// forward_algo = forward_best_algo;
修正したら保存してsetup.pyを実行。
sudo python3 setup.py install
30分ぐらいかかるので気長に待つ。
これでdlibを使用出来るようになった。
参考
Build a Hardware-based Face Recognition System for 0 with the Nvidia Jetson Nano and Python
Using Python 3.6, OpenCV, Dlib and the face_recognition module
issues with dlib library
Please find the code below import face_recognition import cv2 import os import numpy as np import dlib face_locations = ...
コメント