diff --git a/train-eval/Dockerfile b/train-eval/Dockerfile
index e860450..5716b93 100644
--- a/train-eval/Dockerfile
+++ b/train-eval/Dockerfile
@@ -5,6 +5,7 @@ RUN apt install python3-pip -y
 RUN apt install unzip -y
 RUN apt install git -y
 
+RUN pip install --user numpy
 RUN pip install --user pandas
 RUN pip install --user torch
 RUN pip install --user keras