From de4ab1c0273f7caa1cca36eebc235ea283558561 Mon Sep 17 00:00:00 2001 From: Jirka Borovec Date: Thu, 2 Jun 2022 14:54:13 +0200 Subject: [PATCH] update NGC docker (#13136) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit * update docker * Apply suggestions from code review Co-authored-by: Akihiro Nitta Co-authored-by: Carlos Mocholí --- dockers/nvidia/Dockerfile | 15 +++++++-------- dockers/release/Dockerfile | 2 +- 2 files changed, 8 insertions(+), 9 deletions(-) diff --git a/dockers/nvidia/Dockerfile b/dockers/nvidia/Dockerfile index 792835c574..6848f6cf34 100644 --- a/dockers/nvidia/Dockerfile +++ b/dockers/nvidia/Dockerfile @@ -12,7 +12,7 @@ # See the License for the specific language governing permissions and # limitations under the License. -ARG PYTORCH_VERSION=21.11 +ARG PYTORCH_VERSION=22.04 # https://docs.nvidia.com/deeplearning/frameworks/pytorch-release-notes FROM nvcr.io/nvidia/pytorch:${PYTORCH_VERSION}-py3 @@ -37,20 +37,19 @@ RUN \ cd .. ; \ fi && \ # save the examples - mv pytorch-lightning/_notebooks notebooks && \ + mv pytorch-lightning/_notebooks/.notebooks/ notebooks && \ mv pytorch-lightning/pl_examples . && \ # Installations \ pip install -q fire && \ - python ./pytorch-lightning/.actions/assistant.py requirements_prune_pkgs horovod --req_files ./pytorch-lightning/requirements/extra.txt && \ - pip install "Pillow>=8.2, !=8.3.0" "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.2, !=8.3.0" "cryptography>=3.4" "py>=1.10" --no-cache-dir && \ + pip install ./pytorch-lightning["extra","loggers","strategies"] --no-cache-dir && \ + pip install -r ./pytorch-lightning/requirements/examples.txt --no-cache-dir && \ rm -rf pytorch-lightning && \ - pip install jupyterlab[all] -U && \ pip list +RUN pip install jupyterlab[all] -U + RUN pip install lightning-grid -U && \ pip install "py>=1.10" "protobuf>=3.15.6" --upgrade-strategy only-if-needed diff --git a/dockers/release/Dockerfile b/dockers/release/Dockerfile index f4083f2dd4..a0ba3a4a41 100644 --- a/dockers/release/Dockerfile +++ b/dockers/release/Dockerfile @@ -36,7 +36,7 @@ RUN \ mv pytorch-lightning-*/ pytorch-lightning ; \ rm *.zip ; \ fi && \ - pip install ./pytorch-lightning["extra"] --no-cache-dir && \ + pip install ./pytorch-lightning["extra","loggers","strategies"] --no-cache-dir && \ rm -rf pytorch-lightning RUN python --version && \