diff --git a/Dockerfile b/Dockerfile index 7dbc997..39183ef 100644 --- a/Dockerfile +++ b/Dockerfile @@ -8,6 +8,7 @@ RUN apt install -y curl RUN pip3 install --user wget RUN pip3 install torch torchvision torchaudio RUN pip3 install sacred && pip3 install GitPython && pip3 install pymongo +RUN pip3 install dvc WORKDIR /app