Skip to content

Support CUDA wheels #466

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

Merged
merged 7 commits into from
Nov 25, 2025
Merged

Support CUDA wheels #466

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
7 commits
Select commit Hold shift + click to select a range
57dfac1
add support for CUDA wheels
leofang Oct 26, 2025
2c09529
fix fat finger
leofang Oct 26, 2025
ed0ab17
Fix formatting
leofang Oct 26, 2025
4ff81e0
another fat finger...
leofang Oct 26, 2025
b0ebca1
change [all] to [ctk]
leofang Oct 27, 2025
6f9fdd6
move dep injection to the main repo
leofang Oct 29, 2025
9a6ce98
Merge branch 'main' into cuda_wheels
leofang Nov 25, 2025
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
12 changes: 7 additions & 5 deletions dist.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -401,13 +401,14 @@ def build_linux(
WHEEL_LINUX_CONFIGS[cuda_version]['system_packages']

if kind == 'cuda':
long_description_tmpl = WHEEL_LONG_DESCRIPTION_CUDA
long_description = WHEEL_LONG_DESCRIPTION_CUDA.format(
version=platform_version,
wheel_suffix=''.join(platform_version.split('.')))
elif kind == 'rocm':
long_description_tmpl = WHEEL_LONG_DESCRIPTION_ROCM
long_description = WHEEL_LONG_DESCRIPTION_ROCM.format(
version=platform_version)
else:
raise AssertionError('Unreachable')
long_description = long_description_tmpl.format(
version=platform_version)

# Rename wheels to manylinux.
asset_name = wheel_name(
Expand Down Expand Up @@ -611,7 +612,8 @@ def build_windows(
'platform_version', cuda_version)
package_name = WHEEL_WINDOWS_CONFIGS[cuda_version]['name']
long_description = WHEEL_LONG_DESCRIPTION_CUDA.format(
version=platform_version)
version=platform_version,
wheel_suffix=''.join(platform_version.split('.')))
asset_name = wheel_name(
package_name, version, python_version, 'win_amd64')
asset_dest_name = asset_name
Expand Down
7 changes: 5 additions & 2 deletions dist_config.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -309,10 +309,13 @@ class _WheelWindowsConfig(TypedDict):


# Long description of the CUDA wheel package in reST syntax.
# `{version}` will be replaced by the CUDA version (e.g., `9.0`).
# `{version}` and `{wheel_suffix}` will be replaced by the CUDA version (e.g., `9.0`).
WHEEL_LONG_DESCRIPTION_CUDA = _long_description_header + '''\
This is a CuPy wheel (precompiled binary) package for CUDA {version}.
You need to install `CUDA Toolkit {version} <https://developer.nvidia.com/cuda-toolkit-archive>`_ to use these packages.
You need to install `CUDA Toolkit {version} <https://developer.nvidia.com/cuda-toolkit-archive>`_ locally to use these packages.
Alternatively, you can install this package together with all needed CUDA components from PyPI by passing the ``[ctk]`` tag::

$ pip install cupy-cuda{wheel_suffix}[ctk]

If you have another version of CUDA, or want to build from source, refer to the `Installation Guide <https://docs.cupy.dev/en/latest/install.html>`_ for instructions.
''' # NOQA
Expand Down