Skip to content

Conversation

chtruong814
Copy link
Contributor

Bump to pytorch 25.09

  • Also bump uv version
  • Exclude additional nvidia packages that only get installed with the open source pytorch wheel

Signed-off-by: Charlie Truong <chtruong@nvidia.com>
@skyw
Copy link
Contributor

skyw commented Oct 17, 2025

The error looks like a triton version issue. Does ci container built installed triton instead of using the one coming with container?
The upstream triton 3.4.0 doesn't work with CUDA 13, but the one in our 25.09 container does.

@chtruong814
Copy link
Contributor Author

@skyw yeah that's likely the issue. I'll try removing the install of triton in this case since we are bumping to newer pytorch container.

Signed-off-by: Charlie Truong <chtruong@nvidia.com>
@skyw skyw merged commit cf9909b into main Oct 21, 2025
14 checks passed
@skyw skyw deleted the chtruong/update-pytorch-25.09 branch October 21, 2025 03:36
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants