Skip to content

EPIC: Add Nix / nixpkgs support for CUDA Python #1935

@rparolin

Description

@rparolin

Summary

Scope and land first-class support for installing and using cuda-python
(and its sub-packages: cuda.bindings, cuda.core, cuda.pathfinder) via
the Nix package manager / nixpkgs.

Today there is no tracked story for Nix users. This EPIC is a placeholder
for the team to break down and scope the work.

Motivation

  • Nix/NixOS users currently have to roll their own derivations to consume
    cuda-python alongside the CUDA Toolkit that nixpkgs already ships
    (see https://search.nixos.org/packages).
  • A reproducible, upstream-blessed path lowers friction for reproducible
    research, HPC, and CI environments that standardize on Nix.
  • Aligns with the cuda.pathfinder direction of supporting many install
    layouts (EPIC: Path finder for CUDA components #451) — Nix store paths are yet another layout to locate
    headers/libs from.

Open questions / areas to scope

  • How can we align Nix work with CCCLs effort?
  • How does cuda.pathfinder locate components when CUDA lives under
    /nix/store/...-cudatoolkit-*/? Confirm discovery works or add a
    Nix-aware search path (ties to EPIC: Path finder for CUDA components #451).
  • Cython build requirements for cuda.bindings under Nix (CUDA version
    pinning, stub libraries, LD_LIBRARY_PATH vs rpath patching).
  • CI: add a Nix build job to catch regressions.
  • Documentation: install instructions for Nix/NixOS users.

References

Metadata

Metadata

Assignees

Labels

EPICSoul of a releasefeatureNew feature or requestpackagingAnything related to wheels or Conda packages

Type

No fields configured for Epic.

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions