Commit ff4478e4 authored by Daniël de Kok's avatar Daniël de Kok
Browse files

cudaPackages.cudnn_9_3: init at 9.3.0.75

parent 90d02f00
Loading
Loading
Loading
Loading
+12 −5
Original line number Diff line number Diff line
@@ -38,9 +38,16 @@ finalAttrs: prevAttrs: {
    ++ lists.optionals finalAttrs.passthru.useCudatoolkitRunfile [ final.cudatoolkit ]
    ++ lists.optionals (!finalAttrs.passthru.useCudatoolkitRunfile) [ final.libcublas.lib ];

  # Tell autoPatchelf about runtime dependencies.
  # Tell autoPatchelf about runtime dependencies. *_infer* libraries only
  # exist in CuDNN 8.
  # NOTE: Versions from CUDNN releases have four components.
  postFixup = strings.optionalString (strings.versionAtLeast finalAttrs.version "8.0.5.0") ''
  postFixup =
    strings.optionalString
      (
        strings.versionAtLeast finalAttrs.version "8.0.5.0"
        && strings.versionOlder finalAttrs.version "9.0.0.0"
      )
      ''
        ${meta.getExe' patchelf "patchelf"} $lib/lib/libcudnn.so --add-needed libcudnn_cnn_infer.so
        ${meta.getExe' patchelf "patchelf"} $lib/lib/libcudnn_ops_infer.so --add-needed libcublas.so --add-needed libcublasLt.so
      '';
+15 −0
Original line number Diff line number Diff line
@@ -257,6 +257,21 @@
        url = "https://developer.download.nvidia.com/compute/cudnn/redist/cudnn/linux-x86_64/cudnn-linux-x86_64-8.9.7.29_cuda12-archive.tar.xz";
        hash = "sha256-R1MzYlx+QqevPKCy91BqEG4wyTsaoAgc2cE++24h47s=";
      }
      {
        version = "9.3.0.75";
        minCudaVersion = "12.0";
        maxCudaVersion = "12.6";
        url = "https://developer.download.nvidia.com/compute/cudnn/redist/cudnn/linux-x86_64/cudnn-linux-x86_64-9.3.0.75_cuda12-archive.tar.xz";
        hash = "sha256-PW7xCqBtyTOaR34rBX4IX/hQC73ueeQsfhNlXJ7/LCY=";
      }
      {
        version = "9.3.0.75";
        minCudaVersion = "11.8";
        maxCudaVersion = "11.8";
        url = "https://developer.download.nvidia.com/compute/cudnn/redist/cudnn/linux-x86_64/cudnn-linux-x86_64-9.3.0.75_cuda11-archive.tar.xz";
        hash = "sha256-Bp2ghM02jzn7gw1MTpMYAwZPtl52b0z33y2ko0aiup8";
      }

    ];
  };
}