Uh oh!
There was an error while loading. Please reload this page.
- Notifications
You must be signed in to change notification settings - Fork 321
Switch to cuda-toolkit metapackage for wheel dependencies with streamlined documentation#883
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.
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
Uh oh!
There was an error while loading. Please reload this page.
Changes from all commits
8f1e18a17fda4756c83a96a12072899bdb4cc73b20158f83919256e91efcb786793bc55d4e179374e393cd7c41cFile filter
Filter by extension
Conversations
Uh oh!
There was an error while loading. Please reload this page.
Jump to
Uh oh!
There was an error while loading. Please reload this page.
Diff view
Diff view
There are no files selected for viewing
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -10,7 +10,7 @@ | ||
| * Optionally, NVRTC, nvJitLink, NVVM, and cuFile from CUDA Toolkit 13.x | ||
| ```{note} | ||
| The optional CUDA Toolkit components can be installed via PyPI, Conda, OS-specific package managers, or local installers (as described in the CUDA Toolkit [Windows](https://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/index.html) and [Linux](https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html) Installation Guides). | ||
| The optional CUDA Toolkit components are now installed via the `cuda-toolkit` metapackage from PyPI for improved dependency resolution. Components can also be installed via Conda, OS-specific package managers, or local installers (as described in the CUDA Toolkit [Windows](https://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/index.html) and [Linux](https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html) Installation Guides). | ||
| ``` | ||
| Starting from v12.8.0, `cuda-python` becomes a meta package which currently depends only on `cuda-bindings`; in the future more sub-packages will be added to `cuda-python`. In the instructions below, we still use `cuda-python` as example to serve existing users, but everything is applicable to `cuda-bindings` as well. | ||
| @@ -27,12 +27,14 @@ Install all optional dependencies with: | ||
| pip install -U cuda-python[all] | ||
| ``` | ||
| Where the optional dependencies are: | ||
| Where the optional dependencies include: | ||
| * nvidia-cuda-nvrtc (Provides NVRTC shared library) | ||
| * nvidia-nvjitlink (Provides nvJitLink shared library) | ||
| * nvidia-cuda-nvcc (Provides NVVM shared library) | ||
| * nvidia-cufile (Provides cuFile shared library) | ||
| * `nvidia-cuda-nvrtc` (NVRTC runtime compilation library) | ||
| * `nvidia-nvjitlink` (nvJitLink library) | ||
| * `nvidia-nvvm` (NVVM library) | ||
| * `nvidia-cufile` (cuFile library, Linux only) | ||
| These are now installed through the `cuda-toolkit` metapackage for improved dependency resolution. | ||
| ## Installing from Conda | ||
leofang marked this conversation as resolved.
Uh oh!There was an error while loading. Please reload this page. | ||
| @@ -41,6 +43,15 @@ Where the optional dependencies are: | ||
| $ conda install -c conda-forge cuda-python | ||
| ``` | ||
| ```{note} | ||
| When using conda, the `cuda-version` metapackage can be used to control the versions of CUDA Toolkit components that are installed to the conda environment. | ||
| ``` | ||
| For example: | ||
| ```console | ||
| $ conda install -c conda-forge cuda-python cuda-version=13 | ||
| ``` | ||
| ## Installing from Source | ||
leofang marked this conversation as resolved.
Uh oh!There was an error while loading. Please reload this page. |
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -17,11 +17,11 @@ dependencies are as follows: | ||
| ## Installing from PyPI | ||
| `cuda.core` works with `cuda.bindings` (part of `cuda-python`) 11 or 12. For example with CUDA 12: | ||
| `cuda.core` works with `cuda.bindings` (part of `cuda-python`) 11 or 12. Test dependencies now use the ``cuda-toolkit`` metapackage for improved dependency resolution. For example with CUDA 12: | ||
leofang marked this conversation as resolved.
Uh oh!There was an error while loading. Please reload this page. leofang marked this conversation as resolved.
Uh oh!There was an error while loading. Please reload this page. | ||
| ```console | ||
| $ pip install cuda-core[cu12] | ||
| ``` | ||
| and likewise use `[cu11]` for CUDA 11. | ||
| and likewise use `[cu11]` for CUDA 11, or `[cu13]` for CUDA 13. | ||
| Note that using `cuda.core` with NVRTC installed from PyPI via `pip install` requires | ||
| `cuda.bindings` 12.8.0+ or 11.8.6+. Likewise, with nvJitLink it requires 12.8.0+. | ||
Uh oh!
There was an error while loading. Please reload this page.