demo 1

app.py

@@ -5,9 +5,12 @@ import os;
 import socket;
 import torch
 
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+device = "cuda";
 
+if not torch.cuda.is_available() and device == "cuda":
+    raise RuntimeError("CUDA device unavailable, please use Dockerfile.cpu instead.") 
 
+print("DEVICE: ", device);
 
 def sysinfo(device):
     RandomTensor = torch.randn(1, 2)  # Example audio tensor