| 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 |