Update app.py

app.py

@@ -361,8 +361,16 @@ install_cuda_toolkit()
 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['LD_LIBRARY_PATH'] = "/usr/local/cuda-12.1/lib64:" + os.environ.get('LD_LIBRARY_PATH', '')
+
+    # 验证 libnvrtc.so.12 是否能被加载
+    import ctypes
+    try:
+        ctypes.CDLL("libnvrtc.so.12")
+        print("libnvrtc.so.12 loaded successfully!")
+    except OSError as e:
+        print("Failed to load libnvrtc.so.12:", e)
     cuda_path = find_cuda()
     
     if cuda_path: