PyTorch Hub models default to CUDA:0 if available (#2472)

* PyTorch Hub models default to CUDA:0 if available

* device as string bug fix
This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-03-14 23:16:17 -07:00
1 parent 2d41e70e82
commit 9b11f0c58b
4 files changed
+9 -7

No files matched your search

+1 -1
View File
@@ -79,7 +79,7 @@ def check_git_status():
f"Use 'git pull' to update or 'git clone {url}' to download latest."
else:
s = f'up to date with {url} ✅'
print(s.encode().decode('ascii', 'ignore') if platform.system() == 'Windows' else s)
print(s.encode().decode('ascii', 'ignore') if platform.system() == 'Windows' else s) # emoji-safe
except Exception as e:
print(e)