diff --git a/python/packaging/libs_wheel/setup.py b/python/packaging/libs_wheel/setup.py index 054c565b..663f7cea 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##"] + if "##CUDA_MAJOR##" == "13": + reqs = ["nvidia-cuda-runtime"] + else: + reqs = ["nvidia-cuda-runtime-cu##CUDA_MAJOR##"] return reqs