Update torhc installation

wyhong3103 · Feb 18, 2024 · 6792a48 · 6792a48
1 parent f1a143c
commit 6792a48
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/backend/Dockerfile b/backend/Dockerfile
@@ -1,7 +1,8 @@
-FROM bitnami/pytorch:2.2.0-debian-11-r15
+FROM python:3.10
 RUN sudo apt-get update \
  && sudo apt-get install -y libgl1-mesa-glx libgtk2.0-0 libsm6 libxext6 
 WORKDIR /home/app
+RUN pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 COPY . .

diff --git a/backend/requirements.txt b/backend/requirements.txt
@@ -3,6 +3,4 @@
 Flask==3.0.2
 numpy==1.26.4
 opencv-python==4.9.0.80
-torch==2.2.0
-torchvision==0.17.0
 pytest==8.0.0