diff --git a/Dockerfile b/Dockerfile index fdd1083..f9c9a42 100644 --- a/Dockerfile +++ b/Dockerfile @@ -8,8 +8,9 @@ RUN pip3 install torch torchvision torchaudio RUN apt install -y curl RUN pip3 install --user wget RUN pip3 install sacred && pip3 install GitPython && pip3 install pymongo +RUN pip3 install dvc[ssh] paramikos RUN pip3 install dvc -RUN pip3 install dvc[ssh] paramiko + WORKDIR /app