Update app.py

app.py

@@ -9,9 +9,16 @@ import torchvision.transforms as transforms
 from net.dornet import Net
 from net.dornet_ddp import Net_ddp
 
+
+print("=" * 50)
+print(f"CUDA available: {torch.cuda.is_available()}")
+print(f"torch version: {torch.__version__}")
+print(f"CUDA version: {torch.version.cuda}")
+print(f"mmcv version: {mmcv.__version__}")
+print("=" * 50)
+
 # init
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
-print(device)
 net = Net(tiny_model=False).to(device)
 model_ckpt_map = {
     "RGB-D-D": "./checkpoints/RGBDD.pth",