build: Fix pytorch image ref in Dockerfile.ngc_pytorch (#936)

chtruong814 · web-flow · commit b721703ca5bf · 2025-08-18T16:22:54.000Z
Signed-off-by: Charlie Truong &lt;chtruong@nvidia.com&gt;
diff --git a/docker/Dockerfile.ngc_pytorch b/docker/Dockerfile.ngc_pytorch
@@ -8,7 +8,7 @@
 # Local NeMo RL source override: docker buildx build --build-context nemo-rl=. -f docker/Dockerfile.ngc_pytorch --tag <registry>/nemo-rl:latest --push .
 #
 # If installing new dependencies in the container, then use "uv pip install new-dependency"
-ARG BASE_IMAGE=nvcr.io/nvidia/nemo:25.06-py3
+ARG BASE_IMAGE=nvcr.io/nvidia/pytorch:25.06-py3
 FROM scratch AS nemo-rl
 ARG NRL_GIT_REF=main
 ADD --keep-git-dir=true https://github.com/NVIDIA-NeMo/RL.git#${NRL_GIT_REF} /

Original file line number	Diff line number	Diff line change
`@@ -8,7 +8,7 @@`
`8`	`8`	`# Local NeMo RL source override: docker buildx build --build-context nemo-rl=. -f docker/Dockerfile.ngc_pytorch --tag <registry>/nemo-rl:latest --push .`
`9`	`9`	`#`
`10`	`10`	`# If installing new dependencies in the container, then use "uv pip install new-dependency"`
`11`		`-ARG BASE_IMAGE=nvcr.io/nvidia/nemo:25.06-py3`
	`11`	`+ARG BASE_IMAGE=nvcr.io/nvidia/pytorch:25.06-py3`
`12`	`12`	`FROM scratch AS nemo-rl`
`13`	`13`	`ARG NRL_GIT_REF=main`
`14`	`14`	`ADD --keep-git-dir=true https://github.com/NVIDIA-NeMo/RL.git#${NRL_GIT_REF} /`