From e025dbb87d40802736ae9d60d0601898b3eab6b7 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Fri, 13 Jan 2023 00:14:14 -0600 Subject: [PATCH] fix(build): install torch before other packages in CPU container --- api/Containerfile.cpu.buster | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/api/Containerfile.cpu.buster b/api/Containerfile.cpu.buster index fa9e2f9e..c1fbd886 100644 --- a/api/Containerfile.cpu.buster +++ b/api/Containerfile.cpu.buster @@ -10,11 +10,11 @@ ENV PATH="/onnx-web/onnx_web/bin:$PATH" RUN pip3 install pip --upgrade +RUN pip3 install torch torchvision --extra-index-url https://download.pytorch.org/whl/cpu --no-cache-dir --ignore-installed + COPY requirements.txt /onnx-web/requirements.txt -RUN pip3 install -r requirements.txt - -RUN pip3 install torch torchvision --extra-index-url https://download.pytorch.org/whl/cpu +RUN pip3 install -r requirements.txt --no-cache-dir COPY onnx_web/ /onnx-web/onnx_web/