Jiangsan123 commited on
Commit
426ed82
·
1 Parent(s): 65bb73a

Auto-install svraster CUDA extension from source

Browse files
Files changed (2) hide show
  1. app.py +6 -0
  2. requirements.txt +0 -1
app.py CHANGED
@@ -6,6 +6,12 @@ Created on Mon Oct 6 10:16:31 2025
6
  """
7
 
8
  # Copyright (c) 2025, NVIDIA CORPORATION. All rights reserved.
 
 
 
 
 
 
9
 
10
  import os
11
  import time
 
6
  """
7
 
8
  # Copyright (c) 2025, NVIDIA CORPORATION. All rights reserved.
9
+ import subprocess, os
10
+
11
+ cuda_path = os.path.join(os.path.dirname(__file__), "svraster/cuda")
12
+ if os.path.exists(cuda_path):
13
+ print("🧩 Installing svraster CUDA extension from source...")
14
+ subprocess.run(["pip", "install", cuda_path], check=True)
15
 
16
  import os
17
  import time
requirements.txt CHANGED
@@ -26,4 +26,3 @@ pycolmap
26
  opencv-python-headless
27
 
28
  # Local CUDA extension (editable install)
29
- -e ./svraster/cuda
 
26
  opencv-python-headless
27
 
28
  # Local CUDA extension (editable install)