From 5ffed14fe1a987baf3f7fd7e61620496dcb8b3d5 Mon Sep 17 00:00:00 2001 From: Arsalan Rabdanov Date: Mon, 2 Sep 2024 13:48:39 +0300 Subject: [PATCH] updated tensorflow install in docker --- Dockerfile | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/Dockerfile b/Dockerfile index d85d5bc..f8bb9bb 100644 --- a/Dockerfile +++ b/Dockerfile @@ -28,10 +28,10 @@ COPY ./README.md /app/ # ----------------------------------- # if you plan to use a GPU, you should install the 'tensorflow-gpu' package -# RUN pip install --trusted-host pypi.org --trusted-host pypi.python.org --trusted-host=files.pythonhosted.org tensorflow-gpu +RUN pip install --trusted-host pypi.org --trusted-host pypi.python.org --trusted-host=files.pythonhosted.org tensorflow # if you plan to use face anti-spoofing, then activate this line -# RUN pip install --trusted-host pypi.org --trusted-host pypi.python.org --trusted-host=files.pythonhosted.org torch==2.1.2 +RUN pip install --trusted-host pypi.org --trusted-host pypi.python.org --trusted-host=files.pythonhosted.org torch==2.1.2 # ----------------------------------- # install deepface from pypi release (might be out-of-date) # RUN pip install --trusted-host pypi.org --trusted-host pypi.python.org --trusted-host=files.pythonhosted.org deepface