new

app.py

@@ -48,22 +48,6 @@ def install_cuda_toolkit():
     print("==> finfish install")
 install_cuda_toolkit()
 
-@spaces.GPU
-def check_gpu():
-    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', '')
-    subprocess.run(['nvidia-smi'])  # 测试 CUDA 是否可用
-    # 显式加载 libnvrtc.so.12
-    cuda_lib_path = "/usr/local/cuda-12.1/lib64/libnvrtc.so.12"
-    try:
-        ctypes.CDLL(cuda_lib_path, mode=ctypes.RTLD_GLOBAL)
-        print(f"Successfully preloaded {cuda_lib_path}")
-    except OSError as e:
-        print(f"Failed to preload {cuda_lib_path}: {e}")
-    print(f"torch.cuda.is_available:{torch.cuda.is_available()}")
-check_gpu()
 
 import base64
 import re
@@ -114,6 +98,23 @@ def init_warpper():
     k3d_wrapper = init_wrapper_from_config('./pipeline/pipeline_config/default.yaml')
 init_warpper()
 
+@spaces.GPU
+def check_gpu():
+    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', '')
+    subprocess.run(['nvidia-smi'])  # 测试 CUDA 是否可用
+    # 显式加载 libnvrtc.so.12
+    cuda_lib_path = "/usr/local/cuda-12.1/lib64/libnvrtc.so.12"
+    try:
+        ctypes.CDLL(cuda_lib_path, mode=ctypes.RTLD_GLOBAL)
+        print(f"Successfully preloaded {cuda_lib_path}")
+    except OSError as e:
+        print(f"Failed to preload {cuda_lib_path}: {e}")
+    print(f"torch.cuda.is_available:{torch.cuda.is_available()}")
+check_gpu()
+
 TEMP_MESH_ADDRESS=''
 
 mesh_cache = None