Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2019-02-25 13:50:01 +01:00
parent d2cd49f059
commit f541861533
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
utils/torch_utils.py
View File

@ -20,5 +20,5 @@ def select_device(force_cpu=False):
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 ', torch.cuda.device_count(), ' GPUs')
print('Using %s\n%s' % (device.type, torch.cuda.get_device_properties(0) if cuda else '')) print('Using %s %s\n' % (device.type, torch.cuda.get_device_properties(0) if cuda else ''))
return device return device