diff --git a/.github/workflows/build-and-test.yml b/.github/workflows/build-and-test.yml index a809567fd..b8eb7bb23 100644 --- a/.github/workflows/build-and-test.yml +++ b/.github/workflows/build-and-test.yml @@ -321,7 +321,11 @@ jobs: - name: Ensure cuda-python installable run: | - pip install cuda_python*.whl + if [[ "${{ matrix.local-ctk }}" == 1 ]]; then + pip install cuda_python*.whl + else + pip install $(ls cuda_python*.whl)[all] + fi checks: name: Check job status diff --git a/cuda_python/pyproject.toml b/cuda_python/pyproject.toml index 93c68abcf..9566bcf92 100644 --- a/cuda_python/pyproject.toml +++ b/cuda_python/pyproject.toml @@ -32,7 +32,7 @@ classifiers = [ "Environment :: GPU :: NVIDIA CUDA", "Environment :: GPU :: NVIDIA CUDA :: 12", ] -dynamic = ["version", "dependencies"] +dynamic = ["version", "dependencies", "optional-dependencies"] [project.urls] homepage = "https://nvidia.github.io/cuda-python/" diff --git a/cuda_python/setup.py b/cuda_python/setup.py index 3c422f91b..7cddd8fd1 100644 --- a/cuda_python/setup.py +++ b/cuda_python/setup.py @@ -17,4 +17,7 @@ install_requires=[ f"cuda-bindings~={version}", ], + extras_require={ + "all": [f"cuda-bindings[all]~={version}"], + }, )