fix(build): make sure Torch is installed first in containers

ssube · Mar 18, 2023 · 4de8c16 · 4de8c16
1 parent 226710a
commit 4de8c16
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 4 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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