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

lltcggie · Sep 5, 2020 · c365977 · c365977
1 parent 07daa36
commit c365977
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/common/waifu2x.cpp b/common/waifu2x.cpp
@@ -431,7 +431,7 @@ Waifu2x::eWaifu2xCudaError Waifu2x::can_use_CUDA()
 					{
 						cudaDeviceProp prop;
 						cudaGetDeviceProperties(&prop, 0);
-						if (prop.major >= 2)
+						if (prop.major >= 3 && prop.minor >= 5 || prop.major >= 4)
 							CudaFlag = eWaifu2xCudaError_OK;
 						else
 							CudaFlag = eWaifu2xCudaError_OldDevice;