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

GUIでCompute Capability 2.0以上かのチェック追加

Browse Source
This commit is contained in:
lltcggie 2016-05-07 17:25:24 +09:00
parent 623fb925b2
commit 6c7ee29f3d
3 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
common/waifu2x.cpp
View File

@ -406,7 +406,14 @@ Waifu2x::eWaifu2xCudaError Waifu2x::can_use_CUDA()
if (cudaRuntimeGetVersion(&runtimeVersion) == cudaSuccess)
{
if (runtimeVersion >= MinCudaDriverVersion && driverVersion >= runtimeVersion)
{
cudaDeviceProp prop;
cudaGetDeviceProperties(&prop, 0);
if (prop.major >= 2)
CudaFlag = eWaifu2xCudaError_OK;
else
CudaFlag = eWaifu2xCudaError_OldDevice;
}
else
CudaFlag = eWaifu2xCudaError_OldVersion;
}

1
common/waifu2x.h
View File

@ -46,6 +46,7 @@ public:
eWaifu2xCudaError_OK = 0,
eWaifu2xCudaError_NotFind,
eWaifu2xCudaError_OldVersion,
eWaifu2xCudaError_OldDevice,
};
enum eWaifu2xcuDNNError

3
waifu2x-caffe-gui/MainDialog.cpp
View File

@ -2325,6 +2325,9 @@ void DialogEvent::CheckCUDNN(HWND hWnd, WPARAM wParam, LPARAM lParam, LPVOID lpD
case Waifu2x::eWaifu2xCudaError_OldVersion:
MessageBox(dh, langStringList.GetString(L"MessageCudaOldVersionError").c_str(), langStringList.GetString(L"MessageTitleResult").c_str(), MB_OK | MB_ICONERROR);
return;
case Waifu2x::eWaifu2xCudaError_OldDevice:
MessageBox(dh, langStringList.GetString(L"MessageCudaOldDeviceError").c_str(), langStringList.GetString(L"MessageTitleResult").c_str(), MB_OK | MB_ICONERROR);
return;
}
switch (Waifu2x::can_use_cuDNN())