Update app.py

app.py

@@ -243,7 +243,11 @@ def images_to_video(images, output_path, fps=30):
 ###############################################################################
 # Configuration.
 ###############################################################################
-
+install_cuda_toolkit()
+os.environ['CUDA_HOME'] = '/usr/local/cuda-12.1'
+os.environ['PATH'] += ':/usr/local/cuda-12.1/bin'
+# os.environ['LD_LIBRARY_PATH'] += ':/usr/local/cuda-12.1/lib64'
+os.environ['LD_LIBRARY_PATH'] = "/usr/local/cuda-12.1/lib64:" + os.environ.get('LD_LIBRARY_PATH', '')
 seed_everything(0)
 
 config_path = 'configs/PRM_inference.yaml'
@@ -356,21 +360,18 @@ def make_mesh(mesh_fpath, planes):
 
     return mesh_fpath, mesh_glb_fpath
 
-install_cuda_toolkit()
+
 @spaces.GPU
 def make3d(images):
-    os.environ['CUDA_HOME'] = '/usr/local/cuda-12.1'
-    os.environ['PATH'] += ':/usr/local/cuda-12.1/bin'
-    # os.environ['LD_LIBRARY_PATH'] += ':/usr/local/cuda-12.1/lib64'
-    os.environ['LD_LIBRARY_PATH'] = "/usr/local/cuda-12.1/lib64:" + os.environ.get('LD_LIBRARY_PATH', '')
-
     # 验证 libnvrtc.so.12 是否能被加载
     import ctypes
+    # 显式加载 libnvrtc.so.12
+    cuda_lib_path = "/usr/local/cuda-12.1/lib64/libnvrtc.so.12"
     try:
-        ctypes.CDLL("libnvrtc.so.12")
-        print("libnvrtc.so.12 loaded successfully!")
+        ctypes.CDLL(cuda_lib_path, mode=ctypes.RTLD_GLOBAL)
+        print(f"Successfully preloaded {cuda_lib_path}")
     except OSError as e:
-        print("Failed to load libnvrtc.so.12:", e)
+        print(f"Failed to preload {cuda_lib_path}: {e}")
     cuda_path = find_cuda()
     
     if cuda_path: