Update stamp_processing/module/yolov5/yolo_utils/torch_utils.py

stamp_processing/module/yolov5/yolo_utils/torch_utils.py

@@ -9,7 +9,7 @@ import torch.nn.functional as F
 def select_device(device=""):
     cpu = device.lower() == "cpu"
     cuda = not cpu and torch.cuda.is_available()
-    return torch.device("cuda:0" if cuda else "cpu")
+    return torch.device("cuda" if cuda else "cpu")
 
 
 def time_synchronized():