From 9d91b561e82f0f1c0a8b81ad5f70c46dbe0fef2b Mon Sep 17 00:00:00 2001
From: Karol Cyganik <karol.cyganik@wp.pl>
Date: Sun, 14 Apr 2024 22:57:39 +0200
Subject: [PATCH] adjust dockerfile

---
 Dockerfile | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Dockerfile b/Dockerfile
index f836e7d..05bd521 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -8,7 +8,7 @@ RUN apt update
 
 RUN apt install python3 python3-pip -y
 
-RUN pip3 install --no-cache-dir kaggle==1.6.6 matplotlib==3.6.3 scikit-learn==1.2.2 torch==2.0.0 torchvision==0.15.1
+RUN pip3 install --no-cache-dir kaggle==1.6.6 matplotlib==3.6.3 scikit-learn==1.2.2 torch==2.0.0 torchvision==0.15.1 lightning
 
 RUN pip3 install --no-cache-dir Pillow==10.2.0