Update modules/Device/Device.py

modules/Device/Device.py

@@ -89,9 +89,9 @@ def get_torch_device() -> torch.device:
         if is_intel_xpu():
             return torch.device("xpu", torch.xpu.current_device())
         else:
-            try:
+            if is_nvidia():
                 return torch.device(torch.cuda.current_device())
-            except AssertionError:
+            else:
                 return torch.device("cpu")