Update app.py

app.py

@@ -71,16 +71,6 @@ def find_cuda():
 
     return None
 
-
-    
-install_cuda_toolkit()
-
-cuda_path = find_cuda()
-
-if cuda_path:
-    print(f"CUDA installation found at: {cuda_path}")
-else:
-    print("CUDA installation not found")
 # import os
 
 # # 确保 CUDA_HOME 设置正确
@@ -352,6 +342,19 @@ def make_mesh(mesh_fpath, planes):
 
 @spaces.GPU
 def make3d(images):
+    install_cuda_toolkit()
+
+    cuda_path = find_cuda()
+    
+    if cuda_path:
+        print(f"CUDA installation found at: {cuda_path}")
+    else:
+        print("CUDA installation not found")
+    os.environ['CUDA_HOME'] = '/usr/local/cuda'
+    os.environ['PATH'] += ':/usr/local/cuda/bin'
+    os.environ['LD_LIBRARY_PATH'] += ':/usr/local/cuda/lib64'
+    
+    subprocess.run(['nvidia-smi'])  # 测试 CUDA 是否可用
     print(f"CUDA_HOME:{os.environ.get('CUDA_HOME')}")
     print(f"CUDA_HOME exist:{os.path.exists(os.environ.get('CUDA_HOME'))}")
     global model