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 |