cuda9.0和OpenCV 3.2.0不兼容问题

解决方案转自这里

最近升级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
修改以下几行

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>
  • 3
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值