NIXL is a Python API meta-package designed to simplify the installation and usage of NIXL's core functionalities across various CUDA versions. It automatically detects the system's CUDA environment and installs the appropriate `nixl-cudaXXX` sub-package, providing a unified `nixl.core` interface for tensor operations, device management, and more. Current version is 1.0.0, with releases tied to new CUDA variant support.
pip install nixlVerified import paths — ran on the pinned version, not inferred.
This quickstart demonstrates initializing the NIXL environment, retrieving device information, and performing basic tensor operations on the CPU. It also includes a commented-out section showing how to attempt a GPU tensor creation, highlighting the requirement for a functional CUDA environment and a compatible `nixl-cudaXXX` backend.
Check your system's CUDA toolkit version (`nvcc --version`). If issues persist, consider manually uninstalling `nixl` and specific `nixl-cudaXXX` packages, then installing the exact variant (e.g., `pip install nixl-cuda121`) after ensuring your `PATH` and `LD_LIBRARY_PATH` environment variables are correctly set for CUDA.
Always import the core module as `import nixl.core as nixl` and then access elements via `nixl.Tensor`, `nixl.DType`, etc. Alternatively, use specific imports like `from nixl.core import Tensor`.
Ensure your system has a compatible NVIDIA GPU and CUDA Toolkit. Verify `nixl` has successfully installed the correct `nixl-cudaXXX` package. You can explicitly check device availability using `nixl.get_device_info()` or by inspecting the `nixl` package version (e.g., `nixl.__version__` should correspond to an installed `nixl-cudaXXX` package).