cudaPackages.tensorrt: 10.8.0.43 -> 10.9.0.34 (#391261)

pkgs/development/cuda-modules/tensorrt/releases.nix

@@ -127,6 +127,22 @@
         filename = "TensorRT-10.8.0.43.Linux.x86_64-gnu.cuda-12.8.tar.gz";
         hash = "sha256-V31tivU4FTQUuYZ8ZmtPZYUvwusefA6jogbl+vvH1J4=";
       }
+      {
+        version = "10.9.0.34";
+        minCudaVersion = "11.0";
+        maxCudaVersion = "11.8";
+        cudnnVersion = "8.9";
+        filename = "TensorRT-10.9.0.34.Linux.x86_64-gnu.cuda-11.8.tar.gz";
+        hash = "sha256-nQtdgeOIxRA8RsL3ZvQHeBxA4dbJvyWEoFvmSxPaBLA=";
+      }
+      {
+        version = "10.9.0.34";
+        minCudaVersion = "12.0";
+        maxCudaVersion = "12.8";
+        cudnnVersion = "9.7";
+        filename = "TensorRT-10.9.0.34.Linux.x86_64-gnu.cuda-12.8.tar.gz";
+        hash = "sha256-M74OYeO/F3u7yrtIkr8BPwyKxx0r5z8oA4SKOCyxQnI=";
+      }
     ];
   };
 }

pkgs/development/python-modules/tensorrt/default.nix

@@ -10,10 +10,22 @@
 
 let
   pyVersion = "${lib.versions.major python.version}${lib.versions.minor python.version}";
+  buildVersion = lib.optionalString (cudaPackages ? tensorrt) cudaPackages.tensorrt.version;
+  wheelVersion = lib.optionalString (cudaPackages ? tensorrt) (
+    if
+      (builtins.elem buildVersion [
+        "8.6.1.6"
+        "10.3.0.26"
+      ])
+    then
+      builtins.concatStringsSep "." (lib.take 3 (builtins.splitVersion buildVersion))
+    else
+      buildVersion
+  );
 in
 buildPythonPackage rec {
   pname = "tensorrt";
-  version = lib.optionalString (cudaPackages ? tensorrt) cudaPackages.tensorrt.version;
+  version = wheelVersion;
 
   src = cudaPackages.tensorrt.src;
 
@@ -30,7 +42,7 @@ buildPythonPackage rec {
   preUnpack = ''
     mkdir -p dist
     tar --strip-components=2 -xf "$src" --directory=dist \
-      "TensorRT-${version}/python/tensorrt-${version}-cp${pyVersion}-none-linux_x86_64.whl"
+      "TensorRT-${buildVersion}/python/tensorrt-${wheelVersion}-cp${pyVersion}-none-linux_x86_64.whl"
   '';
 
   sourceRoot = ".";