FROM tensorflow/tensorflow:2.4.3-gpu

RUN apt-get update && apt-get install -y \
    libsm6 libxext6 libxrender-dev

COPY requirements.txt .
RUN python -m pip install --no-cache -r requirements.txt -f https://download.pytorch.org/whl/torch_stable.html

COPY . /nightimaging
WORKDIR /nightimaging