From 16bc3b72c333e57f431f42028c0b41115cba476f Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Fri, 28 Dec 2018 20:11:10 +0100
Subject: [PATCH] updates

---
 utils/torch_utils.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/utils/torch_utils.py b/utils/torch_utils.py
index 58bf5ff4..11a09627 100644
--- a/utils/torch_utils.py
+++ b/utils/torch_utils.py
@@ -13,6 +13,7 @@ def init_seeds(seed=0):
     if CUDA_AVAILABLE:
         torch.cuda.manual_seed(seed)
         torch.cuda.manual_seed_all(seed)
+        # torch.cuda.set_device(0)  # OPTIONAL: Set your GPU if multiple available
 
 
 def select_device(force_cpu=False):