修改opencv原码部分参考https://blog.csdn.net/u014613745/article/details/78310916,谢谢大神!
1).找到FindCUDA.cmake文件
找到行
find_cuda_helper_libs(nppi)
改为
-
find_cuda_helper_libs(nppial)
-
find_cuda_helper_libs(nppicc)
-
find_cuda_helper_libs(nppicom)
-
find_cuda_helper_libs(nppidei)
-
find_cuda_helper_libs(nppif)
-
find_cuda_helper_libs(nppig)
-
find_cuda_helper_libs(nppim)
-
find_cuda_helper_libs(nppist)
-
find_cuda_helper_libs(nppisu)
-
find_cuda_helper_libs(nppitc)
2).找到行
set(CUDA_npp_LIBRARY "${CUDA_nppc_LIBRARY};${CUDA_nppi_LIBRARY};${CUDA_npps_LIBRARY}")
改为
set(CUDA_npp_LIBRARY "${CUDA_nppc_LIBRARY};${CUDA_nppial_LIBRARY};${CUDA_nppicc_LIBRARY};${CUDA_nppicom_LIBRARY};${CUDA_nppidei_LIBRARY};${CUDA_nppif_LIBRARY};${CUDA_nppig_LIBRARY};${CUDA_nppim_LIBRARY};${CUDA_nppist_LIBRARY};${CUDA_nppisu_LIBRARY};${CUDA_nppitc_LIBRARY};${CUDA_npps_LIBRARY}")
3).找到行
unset(CUDA_nppi_LIBRARY CACHE)
改为
-
unset(CUDA_nppial_LIBRARY CACHE)
-
unset(CUDA_nppicc_LIBRARY CACHE)
-
unset(CUDA_nppicom_LIBRARY CACHE)
-
unset(CUDA_nppidei_LIBRARY CACHE)
-
unset(CUDA_nppif_LIBRARY CACHE)
-
unset(CUDA_nppig_LIBRARY CACHE)
-
unset(CUDA_nppim_LIBRARY CACHE)
-
unset(CUDA_nppist_LIBRARY CACHE)
-
unset(CUDA_nppisu_LIBRARY CACHE)
-
unset(CUDA_nppitc_LIBRARY CACHE)
4).找到文件OpenCVDetectCUDA.cmake
修改以下几行
-
...
-
set(__cuda_arch_ptx "")
-
if(CUDA_GENERATION STREQUAL "Fermi")
-
set(__cuda_arch_bin "2.0")
-
elseif(CUDA_GENERATION STREQUAL "Kepler")
-
set(__cuda_arch_bin "3.0 3.5 3.7")
-
...
改
-
...
-
set(__cuda_arch_ptx "")
-
if(CUDA_GENERATION STREQUAL "Kepler")
-
set(__cuda_arch_bin "3.0 3.5 3.7")
-
elseif(CUDA_GENERATION STREQUAL "Maxwell")
-
set(__cuda_arch_bin "5.0 5.2")
-
...
5).cuda9中有一个单独的halffloat(cuda_fp16.h)头文件,也应该被包括在opencv的目录里
将头文件cuda_fp16.h添加至 opencv\modules\cudev\include\opencv2\cudev\common.hpp
即在common.hpp中添加
#include <cuda_fp16.h>
cd opencv3.3
mkdir build
cd build
cmake -D CMAKE_BUILD_TYPE=RELEASE -D CMAKE_INSTALL_PREFIX=/usr/local -D CUDA_GENERATION=Kepler ..
make -j8
sudo make install
ok!