diff --git a/Dockerfile b/Dockerfile
index 4d5e992..7ca2cab 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -12,6 +12,7 @@ RUN apt-get install -y python3-pip
 RUN python3 -m pip install pandas
 RUN python3 -m pip install numpy
 RUN python3 -m pip install torch
+RUN python3 -m pip install torchvision
 
 COPY ./zadanie1.py ./
 COPY ./Customers.csv ./