From 54b62f5302b781dce7192eee1f8ba4533fe3ef75 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 4 Mar 2019 17:36:41 +0100 Subject: [PATCH] updates --- utils/torch_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/torch_utils.py b/utils/torch_utils.py index 504aeba2..f98da7fb 100644 --- a/utils/torch_utils.py +++ b/utils/torch_utils.py @@ -18,7 +18,7 @@ def select_device(force_cpu=False): if torch.cuda.device_count() > 1: print('Found %g GPUs' % torch.cuda.device_count()) print('WARNING Multi-GPU Issue: https://github.com/ultralytics/yolov3/issues/21') - # torch.cuda.set_device(0) # OPTIONAL: Set your GPU if multiple available + torch.cuda.set_device(0) # OPTIONAL: Set your GPU if multiple available # # print('Using ', torch.cuda.device_count(), ' GPUs') print('Using %s %s\n' % (device.type, torch.cuda.get_device_properties(0) if cuda else ''))