CUDA 次要版本兼容性

CUDA 次版本兼容性 (MVC) 允许使用比驱动程序支持的CUDA版本更新的CUDA Toolkit版本，前提是Toolkit和驱动程序的主版本相同。例如，通过MVC支持使用CUDA Toolkit 11.5与CUDA驱动程序450（CUDA版本11.0）。

Numba 支持在 Linux 上使用外部 pynvjitlink 包进行 CUDA 12 的 MVC。

Numba 支持在 Linux 上使用外部 cubinlinker 和 ptxcompiler 包对 CUDA 11 进行 MVC，但受以下限制：

  • 不支持存档链接。

  • 协作组不受支持，因为它们需要链接一个存档。

MVC 在 Windows 上不受支持。

安装

CUDA 12

要使用MVC支持，必须安装 pynvjitlink 包。使用conda安装时，请使用：

conda install -c rapidsai pynvjitlink

要使用 pip 安装，请使用 NVIDIA 包索引：

pip install --extra-index-url https://pypi.nvidia.com pynvjitlink-cu12

CUDA 11

要使用MVC支持，必须从适当的渠道安装 cubinlinker 和 ptxcompiler 编译器包。使用conda安装时，请使用：

conda install -c rapidsai -c conda-forge cubinlinker ptxcompiler

要使用 pip 安装，请使用 NVIDIA 包索引：

pip install --extra-index-url https://pypi.nvidia.com ptxcompiler-cu11 cubinlinker-cu11

启用MVC支持

通过设置环境变量来启用MVC支持：

export NUMBA_CUDA_ENABLE_MINOR_VERSION_COMPATIBILITY=1

或者在使用Numba中的任何CUDA功能之前设置一个配置变量：

from numba import config
config.CUDA_ENABLE_MINOR_VERSION_COMPATIBILITY = True

参考文献

关于次版本兼容性的更多信息可以在以下内容中找到：