fix(build): use platform-specific requirements for containers
This commit is contained in:
parent
31054c4ee3
commit
05c628d8a6
|
@ -14,11 +14,9 @@ ENV PATH="/onnx-web/api/onnx_env/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/ /onnx-web/api/requirements/
|
||||
|
||||
COPY requirements.txt /onnx-web/api/requirements.txt
|
||||
|
||||
RUN pip3 install -r requirements.txt --no-cache-dir
|
||||
RUN pip3 install -r requirements/base.txt -r requirements/cpu.txt --no-cache-dir
|
||||
|
||||
COPY onnx_web/ /onnx-web/api/onnx_web/
|
||||
|
||||
|
|
|
@ -16,13 +16,9 @@ ENV PATH="/onnx-web/api/onnx_env/bin:$PATH"
|
|||
|
||||
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.txt --no-cache-dir
|
||||
|
||||
RUN pip3 install onnxruntime-gpu --no-cache-dir
|
||||
RUN pip3 install -r requirements/base.txt -r requirements/nvidia.txt --no-cache-dir
|
||||
|
||||
COPY onnx_web/ /onnx-web/api/onnx_web/
|
||||
|
||||
|
|
|
@ -10,15 +10,9 @@ ENV PATH="/onnx-web/api/onnx_env/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/ /onnx-web/api/requirements/
|
||||
|
||||
RUN pip3 install onnxruntime-directml --force-reinstall
|
||||
|
||||
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
|
||||
RUN pip3 install -r requirements/base.txt -r requirements/amd-windows-nightly.txt --no-cache-dir
|
||||
|
||||
COPY onnx_web/ /onnx-web/api/onnx_web/
|
||||
|
||||
|
|
|
@ -14,15 +14,9 @@ ENV PATH="/onnx-web/api/onnx_env/bin:$PATH"
|
|||
|
||||
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
|
||||
|
||||
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
|
||||
RUN pip3 install -r requirements/base.txt -r requirements/amd-linux.txt --no-cache-dir
|
||||
|
||||
WORKDIR /onnx-web/api
|
||||
|
||||
|
|
Loading…
Reference in New Issue