CV项目调试CUDNN_CONVOLUTION_FWD_SPECIFY_WORKSPACE_LIMIT问题

Posted 2022-07-31 mb62c788fd198da

前言

 

 

error

./src/convolutional_layer.c:153:13: error: ‘CUDNN_CONVOLUTION_FWD_SPECIFY_WORKSPACE_LIMIT’ undeclared (first use in this function)
  153

原因

cudnn8.x里是没有CUDNN_CONVOLUTION_FWD_SPECIFY_WORKSPACE_LIMIT这个宏定义的，而CUDA11.x又不能配套使用cudnn7.x，但是RTX30序列的GPU又必须使用CUDA11.x才能正常跑，感觉进了死胡同。后来找了比较久搜到NVIDIA给出了一个针对cudnn8的解决方案代码，就是修改出错的文件src/convolutional_layer.c的代码，增加针对CUDNN_MAJOR>=8的处理。

 

参考

1. ​​yolo_with_cudnn​​；

2. ​​如何解决pjreddie版darknet不能使用cudnn8编译的问题​​；

完