File tree Expand file tree Collapse file tree 3 files changed +9
-2
lines changed
Expand file tree Collapse file tree 3 files changed +9
-2
lines changed Original file line number Diff line number Diff line change @@ -217,4 +217,8 @@ jobs:
217217
218218 - name : Ensure cuda-python installable
219219 run : |
220- pip install cuda_python*.whl
220+ if [[ "${{ inputs.local-ctk }}" == 1 ]]; then
221+ pip install cuda_python*.whl
222+ else
223+ pip install $(ls cuda_python*.whl)[all]
224+ fi
Original file line number Diff line number Diff line change @@ -32,7 +32,7 @@ classifiers = [
3232 " Environment :: GPU :: NVIDIA CUDA" ,
3333 " Environment :: GPU :: NVIDIA CUDA :: 12" ,
3434]
35- dynamic = [" version" , " dependencies" ]
35+ dynamic = [" version" , " dependencies" , " optional-dependencies " ]
3636
3737[project .urls ]
3838homepage = " https://nvidia.github.io/cuda-python/"
Original file line number Diff line number Diff line change 1717 install_requires = [
1818 f"cuda-bindings~={ version } " ,
1919 ],
20+ extras_require = {
21+ "all" : [f"cuda-bindings[all]~={ version } " ],
22+ },
2023)
You can’t perform that action at this time.
0 commit comments