get torch version

app.py

@@ -29,25 +29,6 @@ from musetalk.utils.preprocessing import get_landmark_and_bbox,read_imgs,coord_p
 from musetalk.utils.blending import get_image
 from musetalk.utils.utils import load_all_model
 
-import subprocess
-
-# Function to install dependencies
-def install_dependencies():
-    try:
-        # Run the pip install commands one by one
-        subprocess.run("pip install --no-cache-dir -U openmim", shell=True, check=True)
-        subprocess.run("mim install mmengine", shell=True, check=True)
-        subprocess.run('mim install "mmcv>=2.0.1"', shell=True, check=True)
-        subprocess.run('mim install "mmdet>=3.1.0"', shell=True, check=True)
-        subprocess.run('mim install "mmpose>=1.1.0"', shell=True, check=True)
-        print("All dependencies installed successfully!")
-    except subprocess.CalledProcessError as e:
-        print(f"An error occurred: {e}")
-        exit(1)
-
-# Call this function before starting the Gradio app
-install_dependencies()
-
 ProjectDir = os.path.abspath(os.path.dirname(__file__))
 CheckpointsDir = os.path.join(ProjectDir, "checkpoints")
 

musetalk/utils/utils.py

@@ -2,17 +2,37 @@ import os
 import cv2
 import numpy as np
 import torch
+import subprocess
 
 current_dir = os.getcwd()
 parent_dir = os.path.dirname(os.path.dirname(current_dir))
 
 os.environ["FFMPEG_PATH"] = f"{parent_dir}/ffmpeg-7.0.2-amd64-static"
 
+# Function to install dependencies
+def install_dependencies():
+    try:
+        # Run the pip install commands one by one
+        subprocess.run("pip install --no-cache-dir -U openmim", shell=True, check=True)
+        subprocess.run("mim install mmengine", shell=True, check=True)
+        subprocess.run('mim install "mmcv>=2.0.1"', shell=True, check=True)
+        subprocess.run('mim install "mmdet>=3.1.0"', shell=True, check=True)
+        subprocess.run('mim install "mmpose>=1.1.0"', shell=True, check=True)
+        print("All dependencies installed successfully!")
+    except subprocess.CalledProcessError as e:
+        print(f"An error occurred: {e}")
+        exit(1)
+
+# Call this function before starting the Gradio app
+install_dependencies()
+
 ffmpeg_path = os.getenv('FFMPEG_PATH')
 if ffmpeg_path is None:
     print("please download ffmpeg-static and export to FFMPEG_PATH. \nFor example: export FFMPEG_PATH=/musetalk/ffmpeg-4.4-amd64-static")
 elif ffmpeg_path not in os.getenv('PATH'):
     print("add ffmpeg to path")
+    print(f"torch version : {torch.__version__}")
+    print(f"torch cuda version : {torch.version.cuda}")
     os.environ["PATH"] = f"{ffmpeg_path}:{os.environ['PATH']}"