test

app.py

@@ -102,7 +102,7 @@ def upsample_features(image, model_option):
         from setuptools import setup, find_packages
         from torch.utils.cpp_extension import BuildExtension, CUDAExtension, CppExtension
 
-        print(subprocess.check_output(['ls', '/usr/local/']).decode())
+        print(subprocess.check_output(['ls', '/usr/local/cuda*']).decode())
         nvcc_path = find_nvcc()
         if nvcc_path:
             print(f"CUDA 'nvcc' found at: {nvcc_path}")