gador/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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

Browse Source 
Signed-off-by: Connor Baker <ConnorBaker01@gmail.com>
This commit is contained in:
Connor Baker
2025-11-25 14:28:00 +00:00
parent d02eb46bfb
commit 2fa699ca66
3 changed files with 45 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkgs/development/cuda-modules/_cuda/manifests/tensorrt/redistrib_10.14.1.json
+36
View File
@@ -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"
}
}
}
}
pkgs/development/cuda-modules/_cuda/manifests/tensorrt/redistrib_10.9.0.json
-36
View File
@@ -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"
}
}
}
}
pkgs/top-level/cuda-packages.nix
+9 -5
View File
@@ -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
{