Skip to content

Add instructions for building CUDA wheels#2896

Merged
henryiii merged 2 commits into
pypa:mainfrom
hcho3:document_cuda
Jun 6, 2026
Merged

Add instructions for building CUDA wheels#2896
henryiii merged 2 commits into
pypa:mainfrom
hcho3:document_cuda

Conversation

@hcho3

@hcho3 hcho3 commented Jun 5, 2026

Copy link
Copy Markdown
Contributor

In the tips and tricks page, add instructions for building CUDA wheels for Linux.

@henryiii

henryiii commented Jun 6, 2026

Copy link
Copy Markdown
Contributor

I tested this in pybind/cmake_example#194, built locally via cibuildwheel on my Mac, and ran on a linux GPU machine. Works great!

@henryiii

henryiii commented Jun 6, 2026

Copy link
Copy Markdown
Contributor

Should we only show the env var version, or maybe have tabs with the config version? It's not perfectly suited to config, as often you probably want multiple CUDA versions. I have ideas for that, but I rather want to see where wheel variants go. (I'm okay with it as is)

@henryiii henryiii merged commit 785d812 into pypa:main Jun 6, 2026
10 checks passed
@hcho3 hcho3 deleted the document_cuda branch June 6, 2026 14:39
henryiii added a commit to henryiii/cibuildwheel that referenced this pull request Jun 9, 2026
The manylinux_cuda project publishes manylinux containers bundling the
CUDA Toolkit (documented in pypa#2896). Add pinned aliases for them so users
can pass a short name (e.g. manylinux_2_28_cuda13_1) to the
manylinux-*-image options, just like the existing manylinux_2_28 alias,
instead of a full registry URL.

These images are only published with a 'latest' tag, and each
architecture has its own repository, so (unlike the PyPA images) they are
pinned by digest. bin/update_docker.py learns to resolve and pin these,
keeping them up to date via `nox -s update_pins`.

Aliases are added for x86_64 and aarch64 (the only arches these images
exist for): manylinux_{2_28,2_34}_cuda{12_9,13_1}.

Assisted-by: ClaudeCode:claude-opus-4.8
Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
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.

2 participants