cuda9.0和OpenCV 3.2.0不兼容问题

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/Mundane_World/article/details/79786272

解决方案转自这里

最近升级Tensorflow到1.6,需要要安装cuda9.0和cudnn7.0(注意:Tensorflow1.6.0最高只支持cudnn7.0.4. 不支持cudnn7.1)。
安装安cuda9.0后,发现不能导入OpenCV了。
所以啊,每次安装/更新cuda都需要重新安装OpenCV。 当然,应该有什么别的好办法,目前还没有查到。。。
在Cmake是后,发现报错,不能成功安装。查了一些资料,发现是因为OpenCV 3.2.0不支持cuda9.0的架构。 需要改一些配置文件。
最后,在这里找到了终极解决方案。
其它的按照之前总结的,一路安装就OK了。

解决方案如下:

1).找到FindCUDA.cmake文件

找到行

view plain copy
find_cuda_helper_libs(nppi)  

改为

view plain copy
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
修改以下几行

扫描二维码关注公众号,回复: 2992155 查看本文章
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>

猜你喜欢

转载自blog.csdn.net/Mundane_World/article/details/79786272