diff --git a/Dockerfile b/Dockerfile
index 4cefcff..e70e496 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -18,6 +18,7 @@ RUN pip3 install sklearn
 RUN pip3 install pymongo
 RUN pip3 install sacred
 RUN pip3 install mlflow
+RUN pip3 install torch
 
 CMD python3 data_expl.py
 CMD python3 nn_train.py
\ No newline at end of file