try

app.py

@@ -4,21 +4,6 @@ import os
 # 获取全部环境变量
 env_vars = os.environ
 
-os.environ["PYTORCH_NVML_BASED_CUDA_CHECK"] = "0"
-import torch
-
-if torch.cuda.is_available():
-    print("CUDA is available. Listing available GPUs:")
-    # 获取并打印GPU数量
-    num_gpus = torch.cuda.device_count()
-    for i in range(num_gpus):
-        print(f"GPU {i}: {torch.cuda.get_device_name(i)}")
-        # 其他相关信息，例如内存
-        print(f"  Memory Allocated: {torch.cuda.memory_allocated(i) / 1024 ** 2:.0f} MB")
-        print(f"  Memory Reserved: {torch.cuda.memory_reserved(i) / 1024 ** 2:.0f} MB")
-else:
-    print("CUDA is not available.")
-
 # 遍历并打印环境变量
 for key, value in env_vars.items():
     print(f"{key}: {value}")