Update stamp_processing/utils.py

stamp_processing/utils.py

@@ -19,7 +19,7 @@ def select_device(device: str = "") -> torch_device:
     """Return a torch.device instance"""
     cpu = device.lower() == "cpu"
     cuda = not cpu and torch.cuda.is_available()
-    return torch_device("cuda" if cuda else "cpu")
+    return torch_device("cuda:1" if cuda else "cpu")
 
 
 def load_yolo_model(weight_path: str, device: str) -> Tuple[Module, int]: