diff --git a/Dockerfile b/Dockerfile
index b09f625..2e4aa2e 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -4,7 +4,7 @@ RUN apt update && apt install -y python3-pip
 RUN apt install unzip
 RUN apt install bc
 
-RUN pip3 install kaggle pandas scikit-learn
+RUN pip3 install kaggle pandas scikit-learn torch
 
 WORKDIR /app