Update evaluate.py

evaluate.py

@@ -254,11 +254,26 @@ def eval(test, standard, tmpdir):
     return sum(scores) / len(scores), final_merged_intervals, comments
 
 def install():
-    print("Start Installing mmpose")
-    os.system('mim install mmengine')
-    # os.system('mim install "mmcv>=2.0.1"')
-    os.system('mim install "mmdet>=3.1.0"')
-    os.system('git clone https://github.com/open-mmlab/mmpose.git')
-    os.chdir('mmpose')
-    os.system('pip install -r requirements.txt')
-    os.system('pip install -v -e .')
+    import torch
+    if torch.cuda.is_available():
+        cu_version = torch.version.cuda
+        cu_version = f"cu{cu_version.replace('.', '')}"  # Format it as 'cuXX' (e.g., 'cu113')
+    else:
+        cu_version = "cpu"  # Fallback to CPU if no CUDA is available
+
+    torch_version = torch.__version__.split('+')[0]  # Get PyTorch version without build info
+
+    pip_command = f'pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/{cu_version}/{torch_version}/index.html'
+
+    print("Executing command:", pip_command)
+
+    # # Execute the command using os.system
+    # os.system(pip_command)
+
+    # os.system('mim install mmengine')
+    # # os.system('mim install "mmcv>=2.0.1"')
+    # os.system('mim install "mmdet>=3.1.0"')
+    # os.system('git clone https://github.com/open-mmlab/mmpose.git')
+    # os.chdir('mmpose')
+    # os.system('pip install -r requirements.txt')
+    # os.system('pip install -v -e .')