mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2024-11-24 03:00:14 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fixed CuDNN runtime version check for CuDNN 9+.

Browse Source
This commit is contained in:
Alexander Smorkalov 2024-07-01 17:33:24 +03:00
parent 39a7b3d186
commit 3d74d646d8
1 changed files with 13 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
modules/dnn/src/cuda4dnn/init.hpp
View File

@ -23,8 +23,19 @@ namespace cv { namespace dnn { namespace cuda4dnn {
// Any patch release x.y.z is forward or backward-compatible with applications built against another cuDNN patch release x.y.w (meaning, of the same major and minor version number, but having w!=z).
// cuDNN minor releases beginning with cuDNN 7 are binary backward-compatible with applications built against the same or earlier patch release (meaning, an application built against cuDNN 7.x is binary compatible with cuDNN library 7.y, where y>=x).
// Applications compiled with a cuDNN version 7.y are not guaranteed to work with 7.x release when y > x.
auto cudnn_bversion = cudnnGetVersion();
auto cudnn_major_bversion = cudnn_bversion / 1000, cudnn_minor_bversion = cudnn_bversion % 1000 / 100;
int cudnn_bversion = cudnnGetVersion();
int cudnn_major_bversion = 0, cudnn_minor_bversion = 0;
// CuDNN changed major version multiplier in 9.0
if (cudnn_bversion >= 9*10000)
{
cudnn_major_bversion = cudnn_bversion / 10000;
cudnn_minor_bversion = cudnn_bversion % 10000 / 100;
}
else
{
cudnn_major_bversion = cudnn_bversion / 1000;
cudnn_minor_bversion = cudnn_bversion % 1000 / 100;
}
if (cudnn_major_bversion != CUDNN_MAJOR || cudnn_minor_bversion < CUDNN_MINOR)
{
std::ostringstream oss;