Browse Source

update no GPU warning message

master
Olivier Courtin 4 months ago
parent
commit
0923fb949e
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      neat_eo/tools/info.py

+ 1
- 1
neat_eo/tools/info.py View File

@@ -91,7 +91,7 @@ def main(args):

if not torch.cuda.device_count():
print("========================================")
print("WARNING: No GPU found, neo train won't be usable.")
print("WARNING: No GPU found, neo train and neo predict won't be usable.")
print("========================================")

if torch.cuda.device_count() and vram < 8:


Loading…
Cancel
Save