diff --git a/Dockerfile b/Dockerfile index 58dacec..c1c2f36 100644 --- a/Dockerfile +++ b/Dockerfile @@ -13,6 +13,11 @@ RUN pip3 install torch RUN pip3 install sklearn RUN pip3 install matplotlib RUN pip3 install sacred +RUN pip3 install numpy +RUN pip3 install wheel --no-deps -U +RUN pip3 install GitPython +RUN pip3 install pymongo +RUN apt-get install unzip ARG CUTOFF ENV CUTOFF=${CUTOFF}