apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(build): use platform-specific requirements for containers

This commit is contained in:
Sean Sube 2023-03-05 17:55:44 -06:00
parent 31054c4ee3
commit 05c628d8a6
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
4 changed files with 8 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
api/Containerfile.cpu.buster
View File

@ -14,11 +14,9 @@ ENV PATH="/onnx-web/api/onnx_env/bin:$PATH"
RUN pip3 install pip --upgrade 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/ /onnx-web/api/requirements/
COPY requirements.txt /onnx-web/api/requirements.txt RUN pip3 install -r requirements/base.txt -r requirements/cpu.txt --no-cache-dir
RUN pip3 install -r requirements.txt --no-cache-dir
COPY onnx_web/ /onnx-web/api/onnx_web/ COPY onnx_web/ /onnx-web/api/onnx_web/

8
api/Containerfile.cuda.ubuntu
View File

@ -16,13 +16,9 @@ ENV PATH="/onnx-web/api/onnx_env/bin:$PATH"
RUN pip3 install pip --upgrade RUN pip3 install pip --upgrade
RUN pip3 install torch torchvision --extra-index-url https://download.pytorch.org/whl/cu117 --no-cache-dir --ignore-installed COPY requirements/ /onnx-web/api/requirements/
COPY requirements.txt /onnx-web/api/requirements.txt RUN pip3 install -r requirements/base.txt -r requirements/nvidia.txt --no-cache-dir
RUN pip3 install -r requirements.txt --no-cache-dir
RUN pip3 install onnxruntime-gpu --no-cache-dir
COPY onnx_web/ /onnx-web/api/onnx_web/ COPY onnx_web/ /onnx-web/api/onnx_web/

10
api/Containerfile.directml.buster
View File

@ -10,15 +10,9 @@ ENV PATH="/onnx-web/api/onnx_env/bin:$PATH"
RUN pip3 install pip --upgrade 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/ /onnx-web/api/requirements/
RUN pip3 install onnxruntime-directml --force-reinstall RUN pip3 install -r requirements/base.txt -r requirements/amd-windows-nightly.txt --no-cache-dir
RUN pip3 install --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ORT-Nightly/pypi/simple/ ort-nightly-directml --force-reinstall
COPY requirements.txt /onnx-web/api/requirements.txt
RUN pip3 install -r requirements.txt --no-cache-dir
COPY onnx_web/ /onnx-web/api/onnx_web/ COPY onnx_web/ /onnx-web/api/onnx_web/

10
api/Containerfile.rocm.ubuntu
View File

@ -14,15 +14,9 @@ ENV PATH="/onnx-web/api/onnx_env/bin:$PATH"
RUN pip3 install pip --upgrade RUN pip3 install pip --upgrade
COPY requirements.txt /onnx-web/api/requirements.txt COPY requirements/ /onnx-web/api/requirements/
RUN pip3 install -r requirements.txt --no-cache-dir RUN pip3 install -r requirements/base.txt -r requirements/amd-linux.txt --no-cache-dir
WORKDIR /ort
RUN wget https://download.onnxruntime.ai/onnxruntime_training-1.14.0.dev20230123001%2Brocm54-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
RUN pip3 install /ort/*.whl --ignore-installed
WORKDIR /onnx-web/api WORKDIR /onnx-web/api