Сегодня я покажу вам, как скомпилировать и установить OpenCV с поддержкой технологии Nvidia CUDA, которая позволит вам использовать GPU для ускорения обработки изображений.
Я предполагаю, что у вас уже установлен инструментарий CUDA. Если нет, есть очень хороший учебник, подготовленный Facebook AI Research (FAIR). Просто посмотрите раздел Установка CUDA в Инструкции FAIR.
Установите необходимые пакеты
sudo apt-get update
sudo apt-get install libopencv-dev build-essential checkinstall cmake pkg-config yasm libtiff4-dev libjpeg-dev libjasper-dev libavcodec-dev libavformat-dev libswscale-dev libdc1394-22-dev libxine-dev libgstreamer0.10-dev libgstreamer-plugins-base0.10-dev libv4l-dev python-dev python-numpy libtbb-dev libqt4-dev libgtk2.0-dev libfaac-dev libmp3lame-dev libopencore-amrnb-dev libopencore-amrwb-dev libtheora-dev libvorbis-dev libxvidcore-dev x264 v4l-utils
sudo add-apt-repository ppa:jon-severinsson/ffmpeg sudo apt-get update sudo apt-get install ffmpeg sudo apt-get install frei0r-plugins
Клонировать репозиторий OpenCV
mkdir OpenCV
cd OpenCV
git clone https://github.com/Itseez/opencv.git
Соберите и установите OpenCV
cd opencv mkdir release cd release
cmake -D CMAKE_BUILD_TYPE=RELEASE -D CMAKE_INSTALL_PREFIX=/usr/local -D WITH_TBB=ON -D BUILD_NEW_PYTHON_SUPPORT=ON -D WITH_V4L=ON -D INSTALL_C_EXAMPLES=ON -D INSTALL_PYTHON_EXAMPLES=ON -D BUILD_EXAMPLES=ON -D WITH_QT=ON -D WITH_OPENGL=ON -D ENABLE_FAST_MATH=1 -D CUDA_FAST_MATH=1 -D WITH_CUBLAS=1 ..
Проверьте вывод cmake и убедитесь, что CUDA и CUBLAS включены:
-- Use Cuda: YES (ver 6.5)
-- Use OpenCL: YES
--
-- NVIDIA CUDA
-- Use CUFFT: YES
-- Use CUBLAS: YES
-- USE NVCUVID: NO
-- NVIDIA GPU arch: 11 12 13 20 21 30 35
-- NVIDIA PTX archs: 30
-- Use fast math: YES
Если все правильно, вы можете установить OpenCV:
make sudo make install
Настроить путь поиска библиотеки
echo '/usr/local/lib' | sudo tee -a /etc/ld.so.conf.d/opencv.conf
sudo ldconfig
printf '# OpenCV\nPKG_CONFIG_PATH=$PKG_CONFIG_PATH:/usr/local/lib/pkgconfig\nexport PKG_CONFIG_PATH\n' >> ~/.bashrc
source ~/.bashrc
Используемые ресурсы:
Первоначально опубликовано на сайте blog.aicry.com 14 марта 2015 г.