Update app.py

app.py

@@ -343,19 +343,17 @@ def make_mesh(mesh_fpath, planes):
 install_cuda_toolkit()
 @spaces.GPU
 def make3d(images):
+    os.environ['CUDA_HOME'] = '/usr/local/cuda-11.8'
+    os.environ['PATH'] += ':/usr/local/cuda-11.8/bin'
+    os.environ['LD_LIBRARY_PATH'] += ':/usr/local/cuda-11.8/lib64'
     
-
     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'
-    
-    
+        
     print(f"CUDA_HOME:{os.environ.get('CUDA_HOME')}")
     print(f"CUDA_HOME exist:{os.path.exists(os.environ.get('CUDA_HOME'))}")
     subprocess.run(['nvidia-smi'])  # 测试 CUDA 是否可用