File tree Expand file tree Collapse file tree 2 files changed +14
-2
lines changed Expand file tree Collapse file tree 2 files changed +14
-2
lines changed Original file line number Diff line number Diff line change @@ -120,4 +120,14 @@ target.
120120 ``/usr/local/cuda/include ``. On Windows, the default is
121121 ``$env:CUDA_PATH\include ``.
122122
123+ .. envvar :: NUMBA_CUDA_ENABLE_MINOR_VERSION_COMPATIBILITY
123124
125+ Enable minor version compatibility for the CUDA driver. Requires the
126+ ``cubinlinker `` and ``ptxcompiler `` packages to be installed. Provides minor
127+ version compatibility for driver versions less than 12.0.
128+
129+ .. envvar :: NUMBA_CUDA_ENABLE_PYNVJITLINK
130+
131+ Use ``pynvjitlink `` for minor version compatibility. Requires the ``pynvjitlink ``
132+ package to be installed. Provides minor version compatibility for driver versions
133+ greater than 12.0.
Original file line number Diff line number Diff line change @@ -65,7 +65,8 @@ MVC support is enabled by setting the environment variable:
6565
6666.. code :: bash
6767
68- export NUMBA_CUDA_ENABLE_MINOR_VERSION_COMPATIBILITY=1
68+ export NUMBA_CUDA_ENABLE_MINOR_VERSION_COMPATIBILITY=1 # CUDA 11
69+ export NUMBA_CUDA_ENABLE_PYNVJITLINK=1 # CUDA 12
6970
7071
7172 or by setting a configuration variable prior to using any CUDA functionality in
7475.. code :: python
7576
7677 from numba import config
77- config.CUDA_ENABLE_MINOR_VERSION_COMPATIBILITY = True
78+ config.CUDA_ENABLE_MINOR_VERSION_COMPATIBILITY = True # CUDA 11
79+ config.CUDA_ENABLE_PYNVJITLINK = True # CUDA 12
7880
7981
8082 References
You can’t perform that action at this time.
0 commit comments