Fix pytorch dependency

wyhong3103 · Feb 19, 2024 · 7f1a879 · 7f1a879
1 parent de395ea
commit 7f1a879
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/backend/requirements.txt b/backend/requirements.txt
@@ -4,5 +4,7 @@ Flask==3.0.2
 numpy==1.26.4
 opencv-python==4.9.0.80
 pytest==8.0.0
-torch==2.2.0 @ https://download.pytorch.org/whl/cpu
-torchvision==0.17.0 @ https://download.pytorch.org/whl/cpu
+-f https://download.pytorch.org/whl/torch_stable.html
+torch==2.2.0+cpu
+-f https://download.pytorch.org/whl/torch_stable.html
+torchvision==0.17.0+cpu