test

app.py

@@ -88,48 +88,13 @@ image_input = gr.Image(label="Choose an image to featurize",
 model_option = gr.Radio(options, value="dino16",
                         label='Choose a backbone to upsample')
 
+print(os.environ['CUDA_HOME'])
 
 @spaces.GPU
 def upsample_features(image, model_option):
-
-    # Check common CUDA installation paths and set the correct one
-    cuda_paths = [
-        '/usr/local/cuda',
-        '/usr/local/cuda-10.0',
-        '/usr/local/cuda-10.1',
-        '/usr/local/cuda-10.2',
-        '/usr/local/cuda-11.0',
-        '/usr/local/cuda-11.1',
-        '/usr/local/cuda-11.2',
-        '/usr/local/cuda-11.3',
-        '/usr/local/cuda-11.4',
-        '/usr/local/cuda-11.5',
-        '/usr/local/cuda-11.6',
-        '/usr/local/cuda-11.7',
-        '/usr/local/cuda-11.8',
-        '/usr/local/cuda-12.0',
-        '/usr/local/cuda-12.1',
-        '/usr/lib/cuda',
-        '/opt/cuda'
-    ]
-
-    cuda_home = None
-    for path in cuda_paths:
-        if os.path.exists(path):
-            cuda_home = path
-            break
-
-    if cuda_home is None:
-        raise EnvironmentError(
-            'CUDA installation not found. Please install CUDA or set CUDA_HOME manually.')
-
-    print(cuda_home)
-
-    # Set the CUDA_HOME environment variable
-    os.environ['CUDA_HOME'] = cuda_home
-    os.environ['PATH'] = os.environ['CUDA_HOME'] + '/bin:' + os.environ['PATH']
-    os.environ['LD_LIBRARY_PATH'] = os.environ['CUDA_HOME'] + \
-        '/lib64:' + os.environ.get('LD_LIBRARY_PATH', '')
+    print(os.environ['CUDA_HOME'])
+    os.environ['PATH'] = '/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin'
+    os.environ['LD_LIBRARY_PATH'] = '/usr/local/nvidia/lib:/usr/local/nvidia/lib64'
 
     # Install the required package from GitHub
     subprocess.check_call(