From 9a047da69b72055b438da0a8af597915fa0d9ad0 Mon Sep 17 00:00:00 2001 From: Adrien Langou Date: Tue, 3 Sep 2024 19:56:50 +0200 Subject: [PATCH] cudaPackages.tensorrt: 8.6 -> 10.3 changelog: https://docs.nvidia.com/deeplearning/tensorrt/release-notes/index.html#rel-10-3-0 --- pkgs/development/cuda-modules/tensorrt/fixup.nix | 3 +-- .../cuda-modules/tensorrt/releases.nix | 16 ++++++++++++++++ 2 files changed, 17 insertions(+), 2 deletions(-) diff --git a/pkgs/development/cuda-modules/tensorrt/fixup.nix b/pkgs/development/cuda-modules/tensorrt/fixup.nix index 42359aedac11..0d52c91320c4 100644 --- a/pkgs/development/cuda-modules/tensorrt/fixup.nix +++ b/pkgs/development/cuda-modules/tensorrt/fixup.nix @@ -99,9 +99,8 @@ finalAttrs: prevAttrs: { cudnn = let desiredName = mkVersionedPackageName "cudnn" package.cudnnVersion; - desiredIsAvailable = final ? desiredName; in - if package.cudnnVersion == null || !desiredIsAvailable then final.cudnn else final.${desiredName}; + if package.cudnnVersion == null || (final ? desiredName) then final.cudnn else final.${desiredName}; }; meta = prevAttrs.meta // { diff --git a/pkgs/development/cuda-modules/tensorrt/releases.nix b/pkgs/development/cuda-modules/tensorrt/releases.nix index a0c29e345a27..976f8fab115b 100644 --- a/pkgs/development/cuda-modules/tensorrt/releases.nix +++ b/pkgs/development/cuda-modules/tensorrt/releases.nix @@ -125,6 +125,22 @@ filename = "TensorRT-8.6.1.6.Linux.x86_64-gnu.cuda-12.0.tar.gz"; hash = "sha256-D4FXpfxTKZQ7M4uJNZE3M1CvqQyoEjnNrddYDNHrolQ="; } + { + version = "10.3.0.26"; + minCudaVersion = "11.0"; + maxCudaVersion = "11.8"; + cudnnVersion = "8.9"; + filename = "TensorRT-10.3.0.26.Linux.x86_64-gnu.cuda-11.8.tar.gz"; + hash = "sha256-1O9TwlUP0eLqTozMs53EefmjriiaHjxb4A4GIuN9jvc="; + } + { + version = "10.3.0.26"; + minCudaVersion = "12.0"; + maxCudaVersion = "12.5"; + cudnnVersion = "8.9"; + filename = "TensorRT-10.3.0.26.Linux.x86_64-gnu.cuda-12.5.tar.gz"; + hash = "sha256-rf8c1avl2HATgGFyNR5Y/QJOW/D8YdSe9LhM047ZkIE="; + } ]; }; }