FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime

ENV TZ=Asia
ARG DEBIAN_FRONTEND=noninteractive

RUN apt-get update && apt-get install -y \
    libpng-dev libjpeg-dev \
    libopencv-dev ffmpeg \
    libgl1-mesa-glx 

COPY requirements.txt .
RUN python -m pip install --upgrade pip
RUN pip install --no-cache -r requirements.txt

COPY . /nightimage
RUN chmod +x /nightimage/run.sh
WORKDIR /nightimage