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

cudaPackages.tensorrt: use 10.13.0 for linux-sbsa on CUDA 12 (#480232)

Browse Source
This commit is contained in:
Connor Baker
2026-01-15 19:28:13 +00:00
committed by GitHub
parent 307ce495f9 ddd283f373
commit 33d8fe302a
2 changed files with 67 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkgs/development/cuda-modules/_cuda/manifests/tensorrt/redistrib_10.13.0.json
+36
View File
@@ -0,0 +1,36 @@
{
"release_date": "2025-07-22",
"release_label": "10.13.0",
"release_product": "tensorrt",
"tensorrt": {
"name": "NVIDIA TensorRT",
"license": "TensorRT",
"version": "10.13.0.35",
"cuda_variant": [
"11",
"12"
],
"linux-sbsa": {
"cuda12": {
"md5": "30ba514ad9252fbaec21d0a336fe5b0a",
"relative_path": "tensorrt/10.13.0/tars/TensorRT-10.13.0.35.Linux.aarch64-gnu.cuda-12.9.tar.gz",
"sha256": "b2f7c7ae6da3588729b7a64825246cc59b643adbe373684fb9c5713262ed46c2",
"size": "4377387760"
}
},
"linux-x86_64": {
"cuda11": {
"md5": "6388feec8573aeced9863a70091aad97",
"relative_path": "tensorrt/10.13.0/tars/TensorRT-10.13.0.35.Linux.x86_64-gnu.cuda-11.8.tar.gz",
"sha256": "4128dd4f40433b957c90f9d04d53311d1636ee046b66a241ab60492f4861de74",
"size": "4788315804"
},
"cuda12": {
"md5": "de94ac8efa527dfd7fa5700f368ebf74",
"relative_path": "tensorrt/10.13.0/tars/TensorRT-10.13.0.35.Linux.x86_64-gnu.cuda-12.9.tar.gz",
"sha256": "899ff39e237f1351fccac7f6891617f68cbab71c13a7d7c703184424bdbbf621",
"size": "6934362378"
}
}
}
}
pkgs/top-level/cuda-packages.nix
+31 -6
View File
@@ -11,9 +11,16 @@ let
manifests = _cuda.lib.selectManifests manifestVersions;
};
# NOTE:
# The manifests are largely the same except for TensorRT:
# - linux-x86_64 is generally the best supported and can use the latest release
# - linux-sbsa (post-Orin Jetson and ARM) comes in second; NVIDIA dropped support for CUDA 12 with 10.13.2 (there is no
# 10.13.1), so we use 10.13.0 for all CUDA 12 releases.
# - linux-aarch64 (pre-Thor Jetson) is historically least supported; we use the latest release available.
cudaPackages_12_6 =
let
inherit (cudaPackages_12_6.backendStdenv) hasJetsonCudaCapability;
inherit (cudaPackages_12_6.backendStdenv) hasJetsonCudaCapability hostPlatform;
in
mkCudaPackages {
cublasmp = "0.6.0";
@@ -29,12 +36,18 @@ let
nvjpeg2000 = "0.9.0";
nvpl = "25.5";
nvtiff = "0.5.1";
tensorrt = if hasJetsonCudaCapability then "10.7.0" else "10.14.1";
tensorrt =
if hasJetsonCudaCapability then
"10.7.0"
else if hostPlatform.isAarch64 then
"10.13.0"
else
"10.14.1";
};
cudaPackages_12_8 =
let
inherit (cudaPackages_12_8.backendStdenv) hasJetsonCudaCapability;
inherit (cudaPackages_12_8.backendStdenv) hasJetsonCudaCapability hostPlatform;
in
mkCudaPackages {
cublasmp = "0.6.0";
@@ -50,12 +63,18 @@ let
nvjpeg2000 = "0.9.0";
nvpl = "25.5";
nvtiff = "0.5.1";
tensorrt = if hasJetsonCudaCapability then "10.7.0" else "10.14.1";
tensorrt =
if hasJetsonCudaCapability then
"10.7.0"
else if hostPlatform.isAarch64 then
"10.13.0"
else
"10.14.1";
};
cudaPackages_12_9 =
let
inherit (cudaPackages_12_9.backendStdenv) hasJetsonCudaCapability;
inherit (cudaPackages_12_9.backendStdenv) hasJetsonCudaCapability hostPlatform;
in
mkCudaPackages {
cublasmp = "0.6.0";
@@ -71,7 +90,13 @@ let
nvjpeg2000 = "0.9.0";
nvpl = "25.5";
nvtiff = "0.5.1";
tensorrt = if hasJetsonCudaCapability then "10.7.0" else "10.14.1";
tensorrt =
if hasJetsonCudaCapability then
"10.7.0"
else if hostPlatform.isAarch64 then
"10.13.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.