File size: 1,202 Bytes
f61a8e6 |
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 |
FROM nvidia/cuda:12.1.1-cudnn8-devel-ubuntu22.04
ENV DEBIAN_FRONTEND=noninteractive
ENV TZ=Europe/Paris
RUN apt-get update
RUN apt-get install -y git git-lfs wget curl make cmake ffmpeg llvm python3 python3-pip
RUN apt-get install -y build-essential xz-utils tk-dev libsm6 libxext6 libgl1-mesa-glx
RUN apt-get install -y libssl-dev zlib1g-dev libbz2-dev libreadline-dev libsqlite3-dev libncursesw5-dev libxml2-dev libxmlsec1-dev libffi-dev liblzma-dev
RUN rm -rf /var/lib/apt/lists/*
RUN git lfs install
RUN useradd -m -u 1000 user
USER user
ENV HOME=/home/user
ENV PATH=/home/user/.local/bin:$PATH
WORKDIR /home/user/app
COPY . /home/user/app
RUN pip install --no-cache-dir addict albumentations addict albumentations basicsr blendmodes einops gradio imageio imageio-ffmpeg invisible-watermark kornia numba omegaconf open_clip_torch prettytable pytorch-lightning safetensors streamlit streamlit-drawable-canvas test-tube timm torchmetrics transformers torch torchvision webdataset xformers yapf
ENV PYTHONPATH=$HOME/app
ENV PYTHONUNBUFFERED=1
ENV GRADIO_ALLOW_FLAGGING=never
ENV GRADIO_NUM_PORTS=1
ENV GRADIO_SERVER_NAME=0.0.0.0
ENV GRADIO_THEME=huggingface
ENV SYSTEM=spaces
RUN python3 app.py |