Skip to content

Commit 7b531ac

Browse files
authored
Fix NVIDIA docker versions (#7834)
1 parent 1da1898 commit 7b531ac

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

dockers/nvidia/Dockerfile

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,10 +44,11 @@ RUN \
4444
pip install -r ./pytorch-lightning/requirements/examples.txt --no-cache-dir --upgrade-strategy only-if-needed && \
4545
pip install ./pytorch-lightning --no-cache-dir && \
4646
rm -rf pytorch-lightning && \
47+
pip install jupyterlab[all] -U && \
4748
pip list
4849

4950
RUN pip install lightning-grid -U && \
50-
pip install "py>=1.10" --upgrade-strategy only-if-needed
51+
pip install "py>=1.10" "protobuf>=3.15.6" --upgrade-strategy only-if-needed
5152

5253
ENV PYTHONPATH="/workspace"
5354

0 commit comments

Comments
 (0)