diff --git a/common/waifu2x.cpp b/common/waifu2x.cpp index d485cb7..fcbff96 100644 --- a/common/waifu2x.cpp +++ b/common/waifu2x.cpp @@ -471,7 +471,7 @@ Waifu2x::eWaifu2xcuDNNError Waifu2x::can_use_cuDNN() cudnnGetVersionType cudnnGetVersionFunc = (cudnnGetVersionType)GetProcAddress(hModule, "cudnnGetVersion"); if (cudnnCreateFunc != nullptr && cudnnDestroyFunc != nullptr && cudnnGetVersionFunc != nullptr) { - if (cudnnGetVersionFunc() >= 3000) + if (cudnnGetVersionFunc() >= CUDNN_REQUIRE_VERION) { cudnnHandle_t h; if (cudnnCreateFunc(&h) == CUDNN_STATUS_SUCCESS) diff --git a/common/waifu2x.h b/common/waifu2x.h index 9647677..8b689d4 100644 --- a/common/waifu2x.h +++ b/common/waifu2x.h @@ -11,7 +11,8 @@ #include #define CUDNN_DLL_NAME "cudnn64_7.dll" -#define CUDNN_REQUIRE_VERION_TEXT "v7" +#define CUDNN_REQUIRE_VERION_TEXT "v7.3" +#define CUDNN_REQUIRE_VERION 7300 namespace caffe