From 1ab3ea21cee46fbb05b8e5f6f5ceea5c5e212518 Mon Sep 17 00:00:00 2001 From: Valerio Mariani Date: Wed, 30 Oct 2024 21:53:32 +0100 Subject: [PATCH 1/2] make base image parametric. default `BASE_IMAGE` is nvcr.io/nvidia/pytorch:24.02-py3 for retro-compatibility Former-commit-id: d5b2def7c029d6f808a97cb062340810e0cd075d --- docker/docker-cuda/Dockerfile | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/docker/docker-cuda/Dockerfile b/docker/docker-cuda/Dockerfile index d03ece88..f4c12462 100644 --- a/docker/docker-cuda/Dockerfile +++ b/docker/docker-cuda/Dockerfile @@ -1,6 +1,5 @@ -# Use the NVIDIA official image with PyTorch 2.3.0 -# https://docs.nvidia.com/deeplearning/frameworks/pytorch-release-notes/rel-24-02.html -FROM nvcr.io/nvidia/pytorch:24.02-py3 +ARG BASE_IMAGE=nvcr.io/nvidia/pytorch:24.02-py3 +FROM ${BASE_IMAGE} # Define environments ENV MAX_JOBS=4 From 5ad86fc71d39b9325c6dee90782dd0d2b12b1ebe Mon Sep 17 00:00:00 2001 From: hoshi-hiyouga Date: Sat, 2 Nov 2024 20:20:26 +0800 Subject: [PATCH 2/2] Update Dockerfile Former-commit-id: 5e8a619d1d20048ad912a1f6d9ccdae5740811bc --- docker/docker-cuda/Dockerfile | 2 ++ 1 file changed, 2 insertions(+) diff --git a/docker/docker-cuda/Dockerfile b/docker/docker-cuda/Dockerfile index f4c12462..34503290 100644 --- a/docker/docker-cuda/Dockerfile +++ b/docker/docker-cuda/Dockerfile @@ -1,3 +1,5 @@ +# Default use the NVIDIA official image with PyTorch 2.3.0 +# https://docs.nvidia.com/deeplearning/frameworks/pytorch-release-notes/index.html ARG BASE_IMAGE=nvcr.io/nvidia/pytorch:24.02-py3 FROM ${BASE_IMAGE}