From 4de8c160c92149a4c4f5c021896082eb3b3119f0 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Fri, 17 Mar 2023 19:41:34 -0500 Subject: [PATCH] fix(build): make sure Torch is installed first in containers --- api/Containerfile.cpu.buster | 4 +++- api/Containerfile.cuda.ubuntu | 4 +++- api/Containerfile.directml.buster | 4 +++- api/Containerfile.rocm.ubuntu | 4 +++- 4 files changed, 12 insertions(+), 4 deletions(-) diff --git a/api/Containerfile.cpu.buster b/api/Containerfile.cpu.buster index cf5af69a..629c6cd7 100644 --- a/api/Containerfile.cpu.buster +++ b/api/Containerfile.cpu.buster @@ -16,7 +16,9 @@ RUN pip3 install pip --upgrade COPY requirements/ /onnx-web/api/requirements/ -RUN pip3 install -r requirements/base.txt -r requirements/cpu.txt --no-cache-dir +RUN pip3 install -r requirements/cpu.txt --no-cache-dir + +RUN pip3 install -r requirements/base.txt --no-cache-dir COPY onnx_web/ /onnx-web/api/onnx_web/ diff --git a/api/Containerfile.cuda.ubuntu b/api/Containerfile.cuda.ubuntu index 9c1e5c22..6f79105e 100644 --- a/api/Containerfile.cuda.ubuntu +++ b/api/Containerfile.cuda.ubuntu @@ -18,7 +18,9 @@ RUN pip3 install pip --upgrade COPY requirements/ /onnx-web/api/requirements/ -RUN pip3 install -r requirements/base.txt -r requirements/nvidia.txt --no-cache-dir +RUN pip3 install -r requirements/nvidia.txt --no-cache-dir + +RUN pip3 install -r requirements/base.txt --no-cache-dir COPY onnx_web/ /onnx-web/api/onnx_web/ diff --git a/api/Containerfile.directml.buster b/api/Containerfile.directml.buster index d4183c8e..4902ca92 100644 --- a/api/Containerfile.directml.buster +++ b/api/Containerfile.directml.buster @@ -12,7 +12,9 @@ RUN pip3 install pip --upgrade COPY requirements/ /onnx-web/api/requirements/ -RUN pip3 install -r requirements/base.txt -r requirements/amd-windows-nightly.txt --no-cache-dir +RUN pip3 install -r requirements/amd-windows-nightly.txt --no-cache-dir + +RUN pip3 install -r requirements/base.txt --no-cache-dir COPY onnx_web/ /onnx-web/api/onnx_web/ diff --git a/api/Containerfile.rocm.ubuntu b/api/Containerfile.rocm.ubuntu index 221faa8f..1d642030 100644 --- a/api/Containerfile.rocm.ubuntu +++ b/api/Containerfile.rocm.ubuntu @@ -16,7 +16,9 @@ RUN pip3 install pip --upgrade COPY requirements/ /onnx-web/api/requirements/ -RUN pip3 install -r requirements/base.txt -r requirements/amd-linux.txt --no-cache-dir +RUN pip3 install -r requirements/amd-linux.txt --no-cache-dir + +RUN pip3 install -r requirements/base.txt --no-cache-dir WORKDIR /onnx-web/api