Skip to content

Investigate JetPack 7.2 / CUDA 13 on AGX Orin 64GB (potential sm_87 wheel unlock) #8

Description

@Vortiago

Summary

JetPack 7.2 (announced 2026-06-01 at GTC Taipei, detailed ~06-12) ships CUDA 13 to the entire Jetson Orin family — Ubuntu 24.04, kernel 6.8. Orin was previously stuck on CUDA 12 throughout the JetPack 6.x line. This is potentially significant for this pipeline, which is pinned to JetPack 6.2.2 / L4T r36.5 and makes several deliberate concessions because of sm_87 (CUDA 12) wheel scarcity:

  • HQQ instead of bitsandbytes for 4-bit (no sm_87 bitsandbytes wheel upstream)
  • eager attention (flash-attn / SDPA unreliable on sm_87)

A CUDA 13 toolchain could change wheel availability and unwind part of docs/versions.md's rationale.

This is INVESTIGATE, not UPGRADE

Open questions to answer before any migration:

  • Is AGX Orin 64GB explicitly supported by JetPack 7.2? (Launch messaging emphasized the 32GB module + "Super Mode" — 64GB support unconfirmed.)
  • Does CUDA 13 on sm_87 actually improve wheel availability — specifically bitsandbytes and flash-attn? If not, the HQQ + eager-attention choices stand regardless.
  • Compatibility of the current pin matrix (transformers, trl, peft, HQQ, torch) against a CUDA 13 / Ubuntu 24.04 base.
  • Migration cost vs. benefit: is there a concrete training/inference win, or is this churn?

If the answers don't show a real win for the 64GB module, document the finding in docs/versions.md and close — keeping the current stack is a valid outcome.

Complexity: moderate (investigation/spike).

References

Filed from a weekly dependency scan (week of 2026-06-22).

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions