gpu usage

app.py

@@ -41,7 +41,11 @@ def cmd(command):
     print(command)
     subprocess.run(shlex.split(command))
 
-@spaces.GPU(duration=150)
+@spaces.GPU
+def cmd_gpu(command):
+    print('gpu:', command)
+    subprocess.run(shlex.split(command))
+
 def process(inputfiles, input_path='demo'):
     if inputfiles:
         frames = natural_sort(inputfiles)
@@ -59,14 +63,9 @@ def process(inputfiles, input_path='demo'):
     for i, frame in enumerate(frames):
         shutil.copy(frame, f"{temp_dir}/{i:04d}.{frame.split('.')[-1]}")
 
-    if torch.cuda.is_available():
-        free_memory, total_memory = torch.cuda.mem_get_info()
-        print(f"Available CUDA memory: {free_memory / (1024 ** 2):.2f} MB")
-        print(f"Total CUDA memory: {total_memory / (1024 ** 2):.2f} MB")
-        
     imgs_path = temp_dir
     output_path = f'./results/{input_path}/output'
-    cmd(f"python dynamic_predictor/launch.py --mode=eval_pose_custom \
+    cmd_gpu(f"python dynamic_predictor/launch.py --mode=eval_pose_custom \
         --pretrained=Kai422kx/das3r \
         --dir_path={imgs_path} \
         --output_dir={output_path} \
@@ -75,22 +74,11 @@ def process(inputfiles, input_path='demo'):
     cmd(f"python utils/rearrange.py --output_dir={output_path}")
     output_path = f'{output_path}_rearranged'
 
-    # Clean up CUDA memory
-    gc.collect()
-    if torch.cuda.is_available():
-        torch.cuda.empty_cache()
-    print(output_path)
-
-    cmd(f"python train_gui.py -s {output_path} -m {output_path} --iter 2000")
-    cmd(f"python render.py -s {output_path} -m {output_path} --iter 2000 --get_video")
-
-    gc.collect()
-    if torch.cuda.is_available():
-        torch.cuda.empty_cache()
-    if torch.cuda.is_available():
-        free_memory, total_memory = torch.cuda.mem_get_info()
-        print(f"Available CUDA memory: {free_memory / (1024 ** 2):.2f} MB")
-        print(f"Total CUDA memory: {total_memory / (1024 ** 2):.2f} MB")
+
+    cmd_gpu(f"python train_gui.py -s {output_path} -m {output_path} --iter 2000")
+    cmd_gpu(f"python render.py -s {output_path} -m {output_path} --iter 2000 --get_video")
+
+
     output_video_path = f"{output_path}/rendered.mp4"
     output_ply_path = f"{output_path}/point_cloud/iteration_2000/point_cloud.ply"
     return  output_video_path, output_ply_path, output_ply_path