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:
1 parent
2d41e70e82
commit
9b11f0c58b
4 files changed
+9
-7
No files matched your search
+1
-1
@@ -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)
|
||||
|
||||
|
||||
Reference in new issue
Block a user