From 9a001fea2218811ec1ce7eb9e8648beca20ede22 Mon Sep 17 00:00:00 2001 From: Jirka Borovec Date: Tue, 1 Jun 2021 12:11:29 +0200 Subject: [PATCH] update NGC docker (#7787) --- dockers/nvidia/Dockerfile | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/dockers/nvidia/Dockerfile b/dockers/nvidia/Dockerfile index fbfd2224a6..8189c5a190 100644 --- a/dockers/nvidia/Dockerfile +++ b/dockers/nvidia/Dockerfile @@ -13,7 +13,7 @@ # limitations under the License. # https://docs.nvidia.com/deeplearning/frameworks/pytorch-release-notes -FROM nvcr.io/nvidia/pytorch:21.04-py3 +FROM nvcr.io/nvidia/pytorch:21.05-py3 LABEL maintainer="PyTorchLightning " @@ -39,14 +39,15 @@ RUN \ # Installations python -c "fname = './pytorch-lightning/requirements/extra.txt' ; lines = [line for line in open(fname).readlines() if not line.startswith('horovod')] ; open(fname, 'w').writelines(lines)" && \ + pip install "Pillow>=8.2" "cryptography>=3.4" "py>=1.10" --no-cache-dir --upgrade-strategy only-if-needed && \ pip install -r ./pytorch-lightning/requirements/extra.txt --no-cache-dir --upgrade-strategy only-if-needed && \ pip install -r ./pytorch-lightning/requirements/examples.txt --no-cache-dir --upgrade-strategy only-if-needed && \ pip install ./pytorch-lightning --no-cache-dir && \ - pip install "Pillow>=8.1" --no-cache-dir --upgrade-strategy only-if-needed && \ rm -rf pytorch-lightning && \ pip list -RUN pip install lightning-grid -U +RUN pip install lightning-grid -U && \ + pip install "py>=1.10" --upgrade-strategy only-if-needed ENV PYTHONPATH="/workspace"