Update install_modules.py

install_modules.py

@@ -133,10 +133,14 @@ if __name__ == "__main__":
     # Установка onnxruntime-gpu из сетевого ресурса
     onnxruntime_index = "https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/"
     install_from_index("onnxruntime-gpu", onnxruntime_index, python_path)
-
+    
+    modules_to_install = ["torch", "torchvision", "torchaudio"]
     pytorch_index = "https://download.pytorch.org/whl/cu124"
-    install_from_index("torch", pytorch_index, python_path)
-    install_from_index("torchvision", pytorch_index, python_path)
-    install_from_index("torchaudio", pytorch_index, python_path)
+    for module in modules_to_install:
+        install_from_index(module, pytorch_index, python_path)
+
+    # install_from_index("torch", pytorch_index, python_path)
+    # install_from_index("torchvision", pytorch_index, python_path)
+    # install_from_index("torchaudio", pytorch_index, python_path)
 
     print(colored("Установка завершена.", 'green'))