Skip to content

Commit

Permalink
Update pytorch base image
Browse files Browse the repository at this point in the history 
There's now a 25.01 release of the pytorch base image, so we should try updating.

Signed-off-by: Peter St. John <pstjohn@nvidia.com>
  • Loading branch information
@pstjohn
pstjohn authored Jan 30, 2025
1 parent b3829e0 commit f037380
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion Dockerfile
View file
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@
# https://gitlab-master.nvidia.com/dl/JoC/nemo-ci/-/blob/main/.gitlab-ci.yml
# We should keep versions in our container up to date to ensure that we get the latest tested perf improvements and
# training loss curves from NeMo.
ARG BASE_IMAGE=nvcr.io/nvidia/pytorch:24.12-py3
ARG BASE_IMAGE=nvcr.io/nvidia/pytorch:25.01-py3

FROM rust:1.82.0 AS rust-env

Expand Down

0 comments on commit f037380

Please sign in to comment.