diff --git a/pkgs/development/cuda-modules/_cuda/manifests/tensorrt/redistrib_10.14.1.json b/pkgs/development/cuda-modules/_cuda/manifests/tensorrt/redistrib_10.14.1.json new file mode 100644 index 000000000000..ed82ea2a666b --- /dev/null +++ b/pkgs/development/cuda-modules/_cuda/manifests/tensorrt/redistrib_10.14.1.json @@ -0,0 +1,36 @@ +{ + "release_date": "2025-11-04", + "release_label": "10.14.1", + "release_product": "tensorrt", + "tensorrt": { + "name": "NVIDIA TensorRT", + "license": "TensorRT", + "version": "10.14.1.48", + "cuda_variant": [ + "12", + "13" + ], + "linux-sbsa": { + "cuda13": { + "md5": "f31d534daa41f517c8c5db8e0478119f", + "relative_path": "tensorrt/10.14.1/tars/TensorRT-10.14.1.48.Linux.aarch64-gnu.cuda-13.0.tar.gz", + "sha256": "3d3e438b49196a9b9ed5c5bc37ec24be9aa74d67d41db0f8fa54b4894aad748a", + "size": "5554126761" + } + }, + "linux-x86_64": { + "cuda12": { + "md5": "d21dd1cb62518f81173150f7bda77311", + "relative_path": "tensorrt/10.14.1/tars/TensorRT-10.14.1.48.Linux.x86_64-gnu.cuda-12.9.tar.gz", + "sha256": "0daa7d5929c78edfbe86b474064d0f82d2064c475cc6be747c5101f1ccc37105", + "size": "7905642926" + }, + "cuda13": { + "md5": "8ef547b7811197572294494551478214", + "relative_path": "tensorrt/10.14.1/tars/TensorRT-10.14.1.48.Linux.x86_64-gnu.cuda-13.0.tar.gz", + "sha256": "c74af67db57f1a0d7e66bb01ab93f1ecda5facac491ca76e680d832f1e035ce6", + "size": "7004299092" + } + } + } +} diff --git a/pkgs/development/cuda-modules/_cuda/manifests/tensorrt/redistrib_10.9.0.json b/pkgs/development/cuda-modules/_cuda/manifests/tensorrt/redistrib_10.9.0.json deleted file mode 100644 index ecf90cbf4328..000000000000 --- a/pkgs/development/cuda-modules/_cuda/manifests/tensorrt/redistrib_10.9.0.json +++ /dev/null @@ -1,36 +0,0 @@ -{ - "release_date": "2025-03-11", - "release_label": "10.9.0", - "release_product": "tensorrt", - "tensorrt": { - "name": "NVIDIA TensorRT", - "license": "TensorRT", - "version": "10.9.0.34", - "cuda_variant": [ - "11", - "12" - ], - "linux-sbsa": { - "cuda12": { - "md5": "e56a9f9d7327c65d9b95996d3008ed44", - "relative_path": "tensorrt/10.9.0/tars/TensorRT-10.9.0.34.Linux.aarch64-gnu.cuda-12.8.tar.gz", - "sha256": "b81ec2a067f67f082c13caec2dcef54385b42a9de6a4ecae6f318aa2d41964f2", - "size": "4123115318" - } - }, - "linux-x86_64": { - "cuda11": { - "md5": "4bad700d0bbfa2c632fdd38319e32690", - "relative_path": "tensorrt/10.9.0/tars/TensorRT-10.9.0.34.Linux.x86_64-gnu.cuda-11.8.tar.gz", - "sha256": "9d0b5d81e388c5103c46c2f766f407781c40e1d6c9bf2584a05be64b13da04b0", - "size": "4422834383" - }, - "cuda12": { - "md5": "ee49e3e6e00b21274907956216b6769f", - "relative_path": "tensorrt/10.9.0/tars/TensorRT-10.9.0.34.Linux.x86_64-gnu.cuda-12.8.tar.gz", - "sha256": "33be0e61e3bf177bbbcabb4892bf013f0c8ac71d2be73f2803848a382cb14272", - "size": "6917032417" - } - } - } -} diff --git a/pkgs/top-level/cuda-packages.nix b/pkgs/top-level/cuda-packages.nix index 3c90f71ed35e..65dac9efe79c 100644 --- a/pkgs/top-level/cuda-packages.nix +++ b/pkgs/top-level/cuda-packages.nix @@ -29,7 +29,7 @@ let nvjpeg2000 = "0.9.0"; nvpl = "25.5"; nvtiff = "0.5.1"; - tensorrt = if hasJetsonCudaCapability then "10.7.0" else "10.9.0"; + tensorrt = if hasJetsonCudaCapability then "10.7.0" else "10.14.1"; }; cudaPackages_12_8 = @@ -50,7 +50,7 @@ let nvjpeg2000 = "0.9.0"; nvpl = "25.5"; nvtiff = "0.5.1"; - tensorrt = if hasJetsonCudaCapability then "10.7.0" else "10.9.0"; + tensorrt = if hasJetsonCudaCapability then "10.7.0" else "10.14.1"; }; cudaPackages_12_9 = @@ -71,12 +71,15 @@ let nvjpeg2000 = "0.9.0"; nvpl = "25.5"; nvtiff = "0.5.1"; - tensorrt = if hasJetsonCudaCapability then "10.7.0" else "10.9.0"; + tensorrt = if hasJetsonCudaCapability then "10.7.0" else "10.14.1"; }; + # NOTE: Thor is supported from CUDA 13.0, so our check needs to capture whether pre-Thor devices were selected. + hasPreThorJetsonCudaCapability = lib.any (lib.flip lib.versionOlder "10.1"); + cudaPackages_13_0 = let - inherit (cudaPackages_13_0.backendStdenv) hasJetsonCudaCapability; + inherit (cudaPackages_13_0.backendStdenv) requestedJetsonCudaCapabilities; in mkCudaPackages { cublasmp = "0.6.0"; @@ -92,7 +95,8 @@ let nvjpeg2000 = "0.9.0"; nvpl = "25.5"; nvtiff = "0.5.1"; - tensorrt = if hasJetsonCudaCapability then "10.7.0" else "10.9.0"; + tensorrt = + if hasPreThorJetsonCudaCapability requestedJetsonCudaCapabilities then "10.7.0" else "10.14.1"; }; in {