test

app.py

@@ -34,9 +34,6 @@ import spaces
 
 # print(adaptive_conv_cuda_impl, adaptive_conv_cpp_impl)
 
-output = subprocess.check_output(['ls', '/']).decode()
-print(output)
-
 def plot_feats(image, lr, hr):
     from featup.util import pca, remove_axes
     assert len(image.shape) == len(lr.shape) == len(hr.shape) == 3
@@ -115,6 +112,13 @@ model_option = gr.Radio(options, value="dino16",
 
 @spaces.GPU
 def upsample_features(image, model_option):
+
+    output = subprocess.check_output(['ls', '/']).decode()
+    print(output)
+    print(torch.version.cuda)
+    cuda_version = subprocess.run(['nvcc', '--version'], stdout=subprocess.PIPE)
+    print(cuda_version.stdout.decode('utf-8'))
+
     from featup.util import norm, unnorm
     models = {o: torch.hub.load("mhamilton723/FeatUp", o) for o in options}