Mirror/waifu2x-caffe
1
0
Fork 0
mirror of https://github.com/lltcggie/waifu2x-caffe.git synced 2025-06-26 21:52:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

cuDNNのバージョンチェックの対象バージョンが古すぎたのを修正 resolved #111

Browse Source
This commit is contained in:
lltcggie 2018-11-23 23:10:54 +09:00
parent bc6281ba7d
commit df0ceaea1e
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
common/waifu2x.cpp
View File

@ -471,7 +471,7 @@ Waifu2x::eWaifu2xcuDNNError Waifu2x::can_use_cuDNN()
cudnnGetVersionType cudnnGetVersionFunc = (cudnnGetVersionType)GetProcAddress(hModule, "cudnnGetVersion"); cudnnGetVersionType cudnnGetVersionFunc = (cudnnGetVersionType)GetProcAddress(hModule, "cudnnGetVersion");
if (cudnnCreateFunc != nullptr && cudnnDestroyFunc != nullptr && cudnnGetVersionFunc != nullptr) if (cudnnCreateFunc != nullptr && cudnnDestroyFunc != nullptr && cudnnGetVersionFunc != nullptr)
{ {
if (cudnnGetVersionFunc() >= 3000) if (cudnnGetVersionFunc() >= CUDNN_REQUIRE_VERION)
{ {
cudnnHandle_t h; cudnnHandle_t h;
if (cudnnCreateFunc(&h) == CUDNN_STATUS_SUCCESS) if (cudnnCreateFunc(&h) == CUDNN_STATUS_SUCCESS)

3
common/waifu2x.h
View File

@ -11,7 +11,8 @@
#include <opencv2/core.hpp> #include <opencv2/core.hpp>
#define CUDNN_DLL_NAME "cudnn64_7.dll" #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 namespace caffe