Skip to content

Bump to torch 2.11#1185

Draft
elynnwu wants to merge 6 commits into
mainfrom
update/bump-to-torch-211
Draft

Bump to torch 2.11#1185
elynnwu wants to merge 6 commits into
mainfrom
update/bump-to-torch-211

Conversation

@elynnwu
Copy link
Copy Markdown
Contributor

@elynnwu elynnwu commented May 19, 2026

Short description of why the PR is needed and how it satisfies those requirements, in sentence form.

Changes:

  • symbol (e.g. fme.core.my_function) or script and concise description of changes or added feature

  • Can group multiple related symbols on a single bullet

  • Tests added

  • If dependencies changed, "deps only" image rebuilt and "latest_deps_only_image.txt" file updated

Resolves # (delete if none)

dependabot Bot and others added 6 commits May 13, 2026 03:52
Bumps [torch](https://github.com/pytorch/pytorch) from 2.7.1 to 2.8.0.
- [Release notes](https://github.com/pytorch/pytorch/releases)
- [Changelog](https://github.com/pytorch/pytorch/blob/main/RELEASE.md)
- [Commits](pytorch/pytorch@v2.7.1...v2.8.0)

---
updated-dependencies:
- dependency-name: torch
  dependency-version: 2.8.0
  dependency-type: direct:production
...

Signed-off-by: dependabot[bot] <support@github.com>
Copy link
Copy Markdown
Contributor

@mcgibbon mcgibbon left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM. One nit, but perhaps it would be best to consider it in a follow-on PR to get this merged.

Comment thread docker/Dockerfile

ENV FORCE_CUDA_EXTENSION=1
ENV CUDA_HOME=/usr/local/cuda
ENV TORCH_CUDA_ARCH_LIST="7.5 8.0 8.9 9.0 10.0+PTX"
Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

nit: Is it possible deduplicate this list against the one in docker/install_torch_harmonics_cuda.sh? Not sure how, but some kind of shared file both reference, if it's possible. Even just a with_cuda_environment.sh <command> script that runs the given command with CUDA_HOME and TORCH_CUDA_ARCH_LIST set.

@mcgibbon
Copy link
Copy Markdown
Contributor

GPU tests are failing with a NCCL error, which is somewhat concerning given the nature of the PR.

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