nftnik commited on
Commit
584121b
verified
1 Parent(s): ae4c3b2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -16,6 +16,11 @@ from nodes import NODE_CLASS_MAPPINGS
16
  from comfy import model_management
17
  import folder_paths
18
 
 
 
 
 
 
19
  # Diret贸rio base e de sa铆da
20
  BASE_DIR = os.path.dirname(os.path.realpath(__file__))
21
  output_dir = os.path.join(BASE_DIR, "output")
 
16
  from comfy import model_management
17
  import folder_paths
18
 
19
+ print("CUDA dispon铆vel:", torch.cuda.is_available())
20
+ print("Quantidade de GPUs:", torch.cuda.device_count())
21
+ if torch.cuda.is_available():
22
+ print("GPU atual:", torch.cuda.get_device_name(0))
23
+
24
  # Diret贸rio base e de sa铆da
25
  BASE_DIR = os.path.dirname(os.path.realpath(__file__))
26
  output_dir = os.path.join(BASE_DIR, "output")