diff --git a/pkgs/development/cuda-modules/tensorrt/releases.nix b/pkgs/development/cuda-modules/tensorrt/releases.nix index 13c7c1a10c32..ea807a8a92ca 100644 --- a/pkgs/development/cuda-modules/tensorrt/releases.nix +++ b/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="; + } ]; }; } diff --git a/pkgs/development/python-modules/tensorrt/default.nix b/pkgs/development/python-modules/tensorrt/default.nix index b69986fdc7e7..2dcdb54035a3 100644 --- a/pkgs/development/python-modules/tensorrt/default.nix +++ b/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 = ".";