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

CUDAデバイスのCCチェックを3.5に更新

Browse Source
This commit is contained in:
lltcggie 2020-09-05 16:30:18 +09:00
parent 07daa3665c
commit c365977a1e
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
common/waifu2x.cpp
View File

@ -431,7 +431,7 @@ Waifu2x::eWaifu2xCudaError Waifu2x::can_use_CUDA()
{ {
cudaDeviceProp prop; cudaDeviceProp prop;
cudaGetDeviceProperties(&prop, 0); cudaGetDeviceProperties(&prop, 0);
if (prop.major >= 2) if (prop.major >= 3 && prop.minor >= 5 || prop.major >= 4)
CudaFlag = eWaifu2xCudaError_OK; CudaFlag = eWaifu2xCudaError_OK;
else else
CudaFlag = eWaifu2xCudaError_OldDevice; CudaFlag = eWaifu2xCudaError_OldDevice;