From 7f1a879dd952d7c2a9573fc22bd9e65b62fc2728 Mon Sep 17 00:00:00 2001 From: wyhong3103 Date: Mon, 19 Feb 2024 21:29:07 +0800 Subject: [PATCH] Fix pytorch dependency --- backend/requirements.txt | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/backend/requirements.txt b/backend/requirements.txt index 8d1eb9a..9897551 100644 --- 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 \ No newline at end of file +-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 \ No newline at end of file