|
|
|
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"] |
