File size: 554 Bytes
27890e3 799c02f 73da713 799c02f 73da713 1e46c0a 799c02f 79b792e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
FROM nvidia/cuda:11.6.0-cudnn8-runtime-ubuntu20.04
ENV GRADIO_SERVER_NAME=0.0.0.0
ENV GRADIO_SERVER_PORT=7860
# Install apt packages
RUN set -ex \
&& apt-get update \
&& DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
python3-dev \
python3-pip \
&& rm -rf /var/cache/apt \
&& rm -rf /var/lib/apt/lists/*
WORKDIR /code
COPY ./ /code
RUN pip3 install --no-cache-dir -r requirements.txt \
&& pip3 install -e . \
&& pip3 uninstall -y onnxruntime \
&& pip3 install onnxruntime-gpu
CMD python3 app2.py
|