NVIDIA NVJITLink
NVIDIA NVJITLink is a Python package that distributes the native NVIDIA NVJITLink shared library (libnvjitlink). This library provides JIT (Just-In-Time) compilation services for NVIDIA GPU code, acting as a low-level component within the NVIDIA ecosystem. It's primarily used as a backend dependency by higher-level Python libraries and frameworks (e.g., Numba, CuPy, PyTorch with NVFuser) that need to compile GPU kernels at runtime. Its version is often tightly coupled with the NVIDIA CUDA Toolkit version. The current version is 13.2.51, and it's updated in conjunction with CUDA Toolkit releases.
Warnings
- gotcha NVJITLink is a low-level backend library. It is not intended for direct use by application developers to write GPU kernels. Its primary function is to provide JIT compilation services to other higher-level frameworks like Numba, CuPy, or PyTorch.
- gotcha The version of `nvidia-nvjitlink` is often tied to a specific NVIDIA CUDA Toolkit version. Mismatches between the installed `nvidia-nvjitlink` package version and the system's CUDA driver or toolkit version can lead to runtime errors or unexpected behavior in dependent libraries.
- gotcha Many Python GPU-accelerated libraries (e.g., `numba`, `cupy`) list `nvidia-nvjitlink` as a dependency and will install it automatically. Directly installing `nvidia-nvjitlink` might not be necessary and could potentially lead to version conflicts if the dependent library requires a specific version.
Install
-
pip install nvidia-nvjitlink
Imports
- __version__
import nvidia_nvjitlink; print(nvidia_nvjitlink.__version__)
Quickstart
import nvidia_nvjitlink
print(f"NVIDIA NVJITLink package version: {nvidia_nvjitlink.__version__}")
# NVJITLink is a backend library; direct high-level API usage is not typical.
# This quickstart confirms the package is installed and its version.