diff --git a/Dockerfile b/Dockerfile index 7dbc997..b831d3f 100644 --- a/Dockerfile +++ b/Dockerfile @@ -8,6 +8,8 @@ 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 +RUN pip3 install 'dvc[ssh]' paramiko WORKDIR /app