Skip to content

build: Bump to NGC PyTorch 26.02 #3474

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
ko3n1g wants to merge 16 commits into
base: main
Choose a base branch
from
+23,078 −23,341
Open

build: Bump to NGC PyTorch 26.02 #3474

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
16 commits
Select commit Hold shift + click to select a range
e30b8ef
build: Bump NGC PyTorch to 26.01
ko3n1g Feb 18, 2026
6352ea9
gitlab bump
ko3n1g Feb 18, 2026
d394d5b
omit weird new file
ko3n1g Feb 18, 2026
57fcd86
test_dcp_checkpoint_save_and_load
ko3n1g Feb 18, 2026
cf32f2f
update golden values
ko3n1g Feb 18, 2026
30f0581
remove marker
ko3n1g Feb 23, 2026
4a38a29
Merge remote-tracking branch 'github/main' into ko3n1g/build/bump-pyt…
ko3n1g Feb 23, 2026
c7f4da3
update golden values
ko3n1g Feb 24, 2026
e1e68de
bump to 26.02
ko3n1g Mar 4, 2026
9f6abea
Merge remote-tracking branch 'github/main' into ko3n1g/build/bump-pyt…
ko3n1g Mar 4, 2026
49b244f
update nightlies
ko3n1g Mar 4, 2026
febfaea
update mr
ko3n1g Mar 4, 2026
2e77eae
install mock
ko3n1g Mar 4, 2026
a0fc3b4
update
ko3n1g Mar 4, 2026
7cd6c7e
Merge branch 'main' into ko3n1g/build/bump-pyt2601
ko3n1g Mar 4, 2026
ae945d9
Merge remote-tracking branch 'github/main' into ko3n1g/build/bump-pyt…
ko3n1g Mar 5, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions .gitlab/stages/01.build.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -64,12 +64,12 @@ test:pre_build_image:
- IMAGE: CI_MCORE_DEV_IMAGE
FILE: Dockerfile.ci.dev
IMAGE_TYPE: dev
BASE_IMAGE: nvcr.io/nvidia/pytorch:25.11-py3
BASE_IMAGE: nvcr.io/nvidia/pytorch:26.02-py3
PLATFORM: amd64
- IMAGE: CI_MCORE_DEV_IMAGE
FILE: Dockerfile.ci.dev
IMAGE_TYPE: dev
BASE_IMAGE: nvcr.io/nvidia/pytorch:25.11-py3
BASE_IMAGE: nvcr.io/nvidia/pytorch:26.02-py3
PLATFORM: arm64
- IMAGE: UTILITY_IMAGE
FILE: Dockerfile.linting
Expand Down
2 changes: 1 addition & 1 deletion docker/.ngc_version.dev
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1 +1 @@
nvcr.io/nvidia/pytorch:25.11-py3
nvcr.io/nvidia/pytorch:26.02-py3
9 changes: 8 additions & 1 deletion pyproject.toml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,12 @@ build-backend = "setuptools.build_meta"
include-package-data = true

[tool.setuptools.packages.find]
include = ["megatron.core", "megatron.core.*", "megatron.training", "megatron.training.*"]
include = [
"megatron.core",
"megatron.core.*",
"megatron.training",
"megatron.training.*",
]

[tool.setuptools.dynamic]
version = { attr = "megatron.core.package_info.__version__" }
Expand Down Expand Up @@ -136,6 +141,7 @@ test = [
"wrapt",
"pytest==8.3.5",
"pytest-mock",
"mock",
"pytest-cov",
"pytest-random-order",
"pytest-asyncio",
Expand Down Expand Up @@ -245,6 +251,7 @@ omit = [
"/opt/megatron-lm/tests/*",
"/opt/megatron-lm/tools/*",
"/usr/local/lib/python3.12/dist-packages/*",
"/opt/megatron-lm/_remote_module_non_scriptable",
]
parallel = true
sigterm = false
Expand Down
Loading
Loading