From 944cd19ce4a27b4527fbb7cbe20a81539fd45c75 Mon Sep 17 00:00:00 2001
From: Mylo <36931363+gitmylo@users.noreply.github.com>
Date: Mon, 12 Feb 2024 21:55:01 +0100
Subject: [PATCH] Update torch_package.py

Related to #214
---
 setup_tools/magicinstaller/requirements/torch_package.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup_tools/magicinstaller/requirements/torch_package.py b/setup_tools/magicinstaller/requirements/torch_package.py
index a3a2e61..f805266 100644
--- a/setup_tools/magicinstaller/requirements/torch_package.py
+++ b/setup_tools/magicinstaller/requirements/torch_package.py
@@ -14,6 +14,6 @@ def is_installed(self):
 
     def install(self):
         # if self.is_windows():
-            return self.install_pip('torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu118', 'PyTorch')
+            return self.install_pip('torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 --extra-index-url https://download.pytorch.org/whl/cu118', 'PyTorch')
         # else:
         #     return self.install_pip('torch torchvision torchaudio', 'PyTorch')