1
0
Fork 0
onnx-web/api/Containerfile.directml.buster

36 lines
943 B
Plaintext
Raw Normal View History

FROM docker.io/python:3.10-buster
WORKDIR /onnx-web/api
RUN pip3 install virtualenv --upgrade
RUN python3 -m venv onnx_env
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
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
COPY onnx_web/ /onnx-web/api/onnx_web/
COPY extras.json logging.yaml params.json /onnx-web/api/
COPY launch.sh /onnx-web/launch.sh
ENV ONNX_WEB_BUNDLE_PATH="/onnx-web/gui"
COPY gui/ /onnx-web/gui/
ENV ONNX_WEB_DEFAULT_PLATFORM="directml"
CMD [ "sh", "-c", "/onnx-web/launch.sh" ]