Announcing the @NixOS/cuda-maintainers team and a call for maintainers!

SergeK · March 16, 2022, 11:50pm

Looking at

cuda-cudnn compatibility tables in cudnn: 8.3.0 -> 8.3.2 by samuela · Pull Request #164338 · NixOS/nixpkgs · GitHub
cudaSupport-conditional hashes in https://github.com/NixOS/nixpkgs/blob/2e80229c8180f05de3d119d726e5b12b05f0a9f9/pkgs/development/python-modules/jaxlib/default.nix#L218

cudaSupport- and platform-conditional hashes in https://github.com/NixOS/nixpkgs/blob/ed8d52174f0995f3baeba694f2edaf73d0598c0f/pkgs/development/python-modules/tensorflow/default.nix#L366

  # .../tensorflow/default.nix
  fetchAttrs = {
    # cudaSupport causes fetch of ncclArchive, resulting in different hashes
    sha256 = if cudaSupport then
      "sha256-+szc2mRoImwijzbj3nw6HmZp3DeRjjPRU5yC+5AEbkg="
    else
      if stdenv.isDarwin then
        "sha256-+bwIzp6t7gRJPcI8B5oyuf9z0AjCAyggUR7x+vv5kFs="
      else
        "sha256-5yOYmeGpJq4Chi55H7iblxyRXVktgnePtpYTPvBs538=";
  };

…one more line of work, in addition to validation in CI, should be reducing the maintenance cost via some sort of update scripts

disclaimer: this has already been brought up in #python:nixos.org