|
|
FROM nvidia/cuda:11.3.1-cudnn8-devel-ubuntu18.04 |
|
|
RUN apt-get update && \ |
|
|
apt-get upgrade -y && \ |
|
|
apt-get install -y --no-install-recommends \ |
|
|
|
|
|
ffmpeg \ |
|
|
apt-get clean && \ |
|
|
rm -rf /var/lib/apt/lists/* |
|
|
RUN useradd -m -u 1000 user |
|
|
USER user |
|
|
ENV HOME=/home/user \ |
|
|
PATH=/home/user/.local/bin:${PATH} |
|
|
WORKDIR ${HOME}/app |
|
|
|
|
|
RUN curl https://pyenv.run | bash |
|
|
ENV PATH=${HOME}/.pyenv/shims:${HOME}/.pyenv/bin:${PATH} |
|
|
ENV PYTHON_VERSION=3.8.8 |
|
|
RUN pyenv install ${PYTHON_VERSION} && \ |
|
|
pyenv global ${PYTHON_VERSION} && \ |
|
|
pyenv rehash && \ |
|
|
pip install --no-cache-dir -U pip setuptools wheel |
|
|
|
|
|
RUN pip install --no-cache-dir -U torch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 |
|
|
COPY --chown=1000 requirements.txt /tmp/requirements.txt |
|
|
RUN pip install --no-cache-dir -U -r /tmp/requirements.txt |
|
|
|
|
|
COPY --chown=1000 . ${HOME}/app |
|
|
RUN ls -a |
|
|
ENV PYTHONPATH=${HOME}/app \ |
|
|
PYTHONUNBUFFERED=1 \ |
|
|
GRADIO_ALLOW_FLAGGING=never \ |
|
|
GRADIO_NUM_PORTS=1 \ |
|
|
GRADIO_SERVER_NAME=0.0.0.0 \ |
|
|
GRADIO_THEME=huggingface \ |
|
|
SYSTEM=spaces |
|
|
CMD ["python", "app.py"] |
