diff --git a/python/packaging/libs_wheel/setup.py b/python/packaging/libs_wheel/setup.py index 054c565b..b3452b70 100644 --- a/python/packaging/libs_wheel/setup.py +++ b/python/packaging/libs_wheel/setup.py @@ -23,7 +23,10 @@ def get_requirements(): - reqs = ["nvidia-cuda-runtime-cu##CUDA_MAJOR##"] + # Use the unified nvidia-cuda-runtime package instead of the deprecated + # CUDA-version-specific packages (nvidia-cuda-runtime-cu12, nvidia-cuda-runtime-cu13, etc.) + # See: https://pypi.org/project/nvidia-cuda-runtime/ + reqs = ["nvidia-cuda-runtime"] return reqs