Update gpu testing function

Author: Fmeson

RawRefinery/application/ModelHandler.py

@@ -13,7 +13,7 @@
 from colour_demosaicing import demosaicing_CFA_Bayer_Malvar2004
 from RawRefinery.application.dng_utils import convert_color_matrix, to_dng
 from RawRefinery.application.postprocessing import match_colors_linear
-from RawRefinery.application.utils import can_use_cuda
+from RawRefinery.application.utils import can_use_gpu
 
 MODEL_REGISTRY = {
     "Tree Net Denoise": {
@@ -184,7 +184,7 @@ def __init__(self):
 
         # Manage devices
         devices = {
-                   "cuda": can_use_cuda(),
+                   "cuda": can_use_gpu(),
                    "mps": torch.backends.mps.is_available(),
                    "cpu": lambda : True
         }

RawRefinery/application/utils.py

@@ -1,15 +1,10 @@
 import torch
 
-def can_use_cuda():
+def can_use_gpu():
     if not torch.cuda.is_available():
-        print("CUDA not available.")
         return False
-    arch = torch.cuda.get_arch_list()
-    major, minor = torch.cuda.get_device_capability()
-
-    if  f"sm_{major}{minor}" in arch:
-        print(f"Found CUDA arch {"sm_{major}{minor}"}. Using Cuda")
+    try:
+        x = torch.zeros(1, device="cuda")
         return True
-    else:
-        print(f"Found CUDA arch {"sm_{major}{minor}"}. Must be in {arch}")
+    except Exception:
         return False