nftnik commited on
Commit
47d3f85
verified
1 Parent(s): f592617

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -0
app.py CHANGED
@@ -22,6 +22,16 @@ output_dir = os.path.join(BASE_DIR, "output")
22
  os.makedirs(output_dir, exist_ok=True)
23
  folder_paths.set_output_directory(output_dir)
24
 
 
 
 
 
 
 
 
 
 
 
25
  # 4. Download de Modelos
26
  def download_models():
27
  models = [
 
22
  os.makedirs(output_dir, exist_ok=True)
23
  folder_paths.set_output_directory(output_dir)
24
 
25
+ # 4. Diagn贸stico CUDA
26
+ print("Python version:", sys.version)
27
+ print("Torch version:", torch.__version__)
28
+ print("CUDA dispon铆vel:", torch.cuda.is_available())
29
+ print("Quantidade de GPUs:", torch.cuda.device_count())
30
+ if torch.cuda.is_available():
31
+ print("GPU atual:", torch.cuda.get_device_name(0))
32
+ else:
33
+ print("GPU n茫o dispon铆vel. Usando CPU.")
34
+
35
  # 4. Download de Modelos
36
  def download_models():
37
  models = [