From df0ceaea1eaa1ccf2266a03f6ee5929f38ece484 Mon Sep 17 00:00:00 2001 From: lltcggie Date: Fri, 23 Nov 2018 23:10:54 +0900 Subject: [PATCH] =?UTF-8?q?cuDNN=E3=81=AE=E3=83=90=E3=83=BC=E3=82=B8?= =?UTF-8?q?=E3=83=A7=E3=83=B3=E3=83=81=E3=82=A7=E3=83=83=E3=82=AF=E3=81=AE?= =?UTF-8?q?=E5=AF=BE=E8=B1=A1=E3=83=90=E3=83=BC=E3=82=B8=E3=83=A7=E3=83=B3?= =?UTF-8?q?=E3=81=8C=E5=8F=A4=E3=81=99=E3=81=8E=E3=81=9F=E3=81=AE=E3=82=92?= =?UTF-8?q?=E4=BF=AE=E6=AD=A3=20resolved=20#111?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- common/waifu2x.cpp | 2 +- common/waifu2x.h | 3 ++- 2 files changed, 3 insertions(+), 2 deletions(-) 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