adjust dockerfile

This commit is contained in:
Karol Cyganik 2024-04-14 22:57:39 +02:00
parent 0fad11aa5c
commit 9d91b561e8
1 changed files with 1 additions and 1 deletions

View File

@ -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