From 8fff742443ad94aa9390bbb6f80ad87b93c7029a Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Fri, 13 Jan 2023 00:38:36 -0600 Subject: [PATCH] fix(build): install torch before other packages in CUDA container --- api/Containerfile.cuda.ubuntu | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/api/Containerfile.cuda.ubuntu b/api/Containerfile.cuda.ubuntu index 6226c127..8f91b9ba 100644 --- a/api/Containerfile.cuda.ubuntu +++ b/api/Containerfile.cuda.ubuntu @@ -12,14 +12,14 @@ 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/cu117 + COPY requirements.txt /onnx-web/requirements.txt RUN pip3 install -r requirements.txt RUN pip3 install onnxruntime-gpu -RUN pip3 install torch torchvision --extra-index-url https://download.pytorch.org/whl/cu117 - COPY onnx_web/ /onnx-web/onnx_web/ ENTRYPOINT [ "flask", "--app=onnx_web.serve", "run", "--host=0.0.0.0" ]