File size: 1,072 Bytes
89999fa f412bec 7a72e74 aea9a02 c541e24 aea9a02 e20eca3 aea9a02 90ed3f1 aea9a02 d303a21 05c5a94 fbe2880 d303a21 aea9a02 cd6e8fe c541e24 41ce9b3 f254a1b 2d51dcb cd6e8fe |
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 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 |
#FROM python:3.11-slim-bullseye
FROM nvidia/cuda:12.3.0-devel-ubuntu22.04
RUN apt-get update && \
apt-get -y install \
git \
python3.10 \
python3-pip \
openmpi-bin \
libopenmpi-dev
COPY . /opt/optimum-nvidia
WORKDIR /opt/optimum-nvidia
RUN pip install --pre --extra-index-url https://pypi.nvidia.com -e '.[quality, tests]' && \
pip uninstall -y optimum-nvidia && \
rm -rf /opt/optimum-nvidia
RUN apt-get update && apt-get upgrade -y && apt-get install -y --no-install-recommends \
python3 \
python3-pip \
ninja-build \
build-essential \
pkg-config \
gnupg2 \
wget
WORKDIR /code
RUN chmod 777 .
COPY ./requirements.txt /code/requirements.txt
RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
RUN useradd -m -u 1000 user
USER user
ENV HOME=/home/user \
PATH=/home/user/.local/bin:$PATH \
TF_ENABLE_ONEDNN_OPTS=0 \
HOST=0.0.0.0 \
PORT=7860 \
ORIGINS=*
WORKDIR $HOME/app
COPY --chown=user . $HOME/app
RUN chmod 777 .
EXPOSE 7860
CMD ["python", "-m", "main"] |