ARG BASE_IMAGE
FROM $BASE_IMAGE

# if needed, can install Ubuntu packages here, e.g.:
RUN apt-get update
RUN apt-get update --fix-missing
RUN apt-get install -y git zip vim gawk libopenmpi-dev

RUN mkdir /smoothing
WORKDIR /smoothing

# install any needed python packages not already provided by the
# base docker image
ARG REQUIREMENTS
ADD $REQUIREMENTS /smoothing/
RUN pip install --upgrade pip
# RUN pip install -r $REQUIREMENTS
RUN pip install scipy pandas statsmodels matplotlib seaborn setGPU tqdm blobfile mpi4py
RUN pip install foolbox \
                torchsde \
                git+https://github.com/RobustBench/robustbench.git

#return to home directory
WORKDIR /smoothing
