downgrade pytorch (#220)

* downgrade pytorch PyTorch 2.2.0 causes some problems with CI. This should temporarily solve the issue until we find a proper fix * pin torchvision version to match torch
AmpereComputingAI · Feb 6, 2024 · 522f0bf · 522f0bf
1 parent a4f1b0d
commit 522f0bf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/utils/setup/install_frameworks.py b/utils/setup/install_frameworks.py
@@ -30,12 +30,12 @@
 try:
     import torch
 except ModuleNotFoundError:
-    subprocess.check_call([sys.executable, "-m", "pip", "install", f"torch"])
+    subprocess.check_call([sys.executable, "-m", "pip", "install", f"torch<2.2"])
     native_frameworks.append("PyTorch")
 try:
     import torchvision
 except ModuleNotFoundError:
-    subprocess.check_call([sys.executable, "-m", "pip", "install", f"torchvision"])
+    subprocess.check_call([sys.executable, "-m", "pip", "install", f"torchvision==0.16.2"])
 
 
 # summary