diff --git a/Dockerfile b/Dockerfile
index 2069193..67cd8dc 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -11,6 +11,7 @@ RUN apt install unzip -y
 RUN pip3 install kaggle
 RUN pip3 install pandas
 RUN pip3 install torch
+RUN pip3 install torchvision
 RUN pip3 install jovian