Update app.py

app.py

@@ -19,10 +19,7 @@ import argparse
 import os
 import logging
 
-try:
-    import cupy
-except:
-    print('import cupy failed!')
+print('torch_version', torch.version.cuda)
 
 import numpy as np
 import torch
@@ -80,6 +77,7 @@ def depth_normal(img,
                 seed,
                 domain):
 
+    print('torch_version', torch.version.cuda)
     seed = int(seed)
     if seed >= 0:
         torch.manual_seed(seed)