diff --git a/Dockerfile b/Dockerfile
index 827ce34..f71f2b4 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -10,6 +10,7 @@ COPY ./tensor.py ./
 COPY requirements.txt ./
 
 RUN pip3 install -r ./requirements.txt
+RUN pip3 install dvc[ssh] paramiko
 
 CMD python3 stats.py
 CMD python3 tensor.py
\ No newline at end of file
diff --git a/Jenkinsfile_dvc b/Jenkinsfile_dvc
index 9b9a89b..62ec9b0 100644
--- a/Jenkinsfile_dvc
+++ b/Jenkinsfile_dvc
@@ -2,7 +2,7 @@
 pipeline {
    agent {
         docker {
-            image 'shroomy/ium:8'
+            image 's434804/ium:0.6'
             args '-v /tmp/mlruns:/tmp/mlruns -v /mlruns:/mlruns '
         }
    }
diff --git a/requirements.txt b/requirements.txt
index 908b24b..29fceed 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,3 +1,4 @@
+dvc==2.3.0
 kaggle==1.5.12
 matplotlib==3.3.4
 mlflow==1.17.0