Commit 0d814c06 authored by Connor Baker's avatar Connor Baker
Browse files

cudaPackages.tensorrt: 10.9.0 -> 10.14.1 (for non-Jetson and Thor) 



Signed-off-by: default avatarConnor Baker <ConnorBaker01@gmail.com>
(cherry picked from commit 2fa699ca)
parent 7eb6ea8e

pkgs/development/cuda-modules/_cuda/manifests/tensorrt/redistrib_10.14.1.json

0 → 100644
+36 −0
Original line number Diff line number Diff line 
{

    "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"

            }

        }

    }

}

pkgs/development/cuda-modules/_cuda/manifests/tensorrt/redistrib_10.9.0.json

deleted100644 → 0
+0 −36
Original line number Diff line number Diff line 
{

    "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"

            }

        }

    }

}

pkgs/top-level/cuda-packages.nix

+9 −5
Original line number Diff line number Diff line
@@ -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

{