From e5d1b86bda7b1ba83d0703041aa83220c8cf4845 Mon Sep 17 00:00:00 2001
From: Jirka <jirka.borovec@seznam.cz>
Date: Fri, 4 Jun 2021 18:39:10 +0200
Subject: [PATCH] fix versions

---
 dockers/nvidia/Dockerfile | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/dockers/nvidia/Dockerfile b/dockers/nvidia/Dockerfile
index 8189c5a190834..528e7561c1e96 100644
--- a/dockers/nvidia/Dockerfile
+++ b/dockers/nvidia/Dockerfile
@@ -44,10 +44,11 @@ RUN \
     pip install -r ./pytorch-lightning/requirements/examples.txt --no-cache-dir --upgrade-strategy only-if-needed && \
     pip install ./pytorch-lightning --no-cache-dir && \
     rm -rf pytorch-lightning && \
+    pip install jupyterlab[all] -U && \
     pip list
 
 RUN pip install lightning-grid -U && \
-    pip install "py>=1.10" --upgrade-strategy only-if-needed
+    pip install "py>=1.10" "protobuf>=3.15.6" --upgrade-strategy only-if-needed
 
 ENV PYTHONPATH="/workspace"