summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorConnor Baker <ConnorBaker01@Gmail.com>2024-09-16 10:38:08 -0700
committerGitHub <noreply@github.com>2024-09-16 10:38:08 -0700
commit3540ebe5534a9e7f38b3c782dba8a03eaeae7303 (patch)
tree15d250b748d6fa3f57895483e678803c0e1eb0d4
parentgtk4-layer-shell: add `wayland-protocols` as a dependency to fix segfault (#3... (diff)
parentcudaPackages.tensorrt: 8.6 -> 10.3 (diff)
downloadnixpkgs-3540ebe5534a9e7f38b3c782dba8a03eaeae7303.tar.gz
cudaPackages.tensorrt: 8.6 -> 10.3 (#339305)
-rw-r--r--pkgs/development/cuda-modules/tensorrt/fixup.nix3
-rw-r--r--pkgs/development/cuda-modules/tensorrt/releases.nix16
2 files changed, 17 insertions, 2 deletions
diff --git a/pkgs/development/cuda-modules/tensorrt/fixup.nix b/pkgs/development/cuda-modules/tensorrt/fixup.nix
index 42359aedac11..0d52c91320c4 100644
--- a/pkgs/development/cuda-modules/tensorrt/fixup.nix
+++ b/pkgs/development/cuda-modules/tensorrt/fixup.nix
@@ -99,9 +99,8 @@ finalAttrs: prevAttrs: {
cudnn =
let
desiredName = mkVersionedPackageName "cudnn" package.cudnnVersion;
- desiredIsAvailable = final ? desiredName;
in
- if package.cudnnVersion == null || !desiredIsAvailable then final.cudnn else final.${desiredName};
+ if package.cudnnVersion == null || (final ? desiredName) then final.cudnn else final.${desiredName};
};
meta = prevAttrs.meta // {
diff --git a/pkgs/development/cuda-modules/tensorrt/releases.nix b/pkgs/development/cuda-modules/tensorrt/releases.nix
index a0c29e345a27..976f8fab115b 100644
--- a/pkgs/development/cuda-modules/tensorrt/releases.nix
+++ b/pkgs/development/cuda-modules/tensorrt/releases.nix
@@ -125,6 +125,22 @@
filename = "TensorRT-8.6.1.6.Linux.x86_64-gnu.cuda-12.0.tar.gz";
hash = "sha256-D4FXpfxTKZQ7M4uJNZE3M1CvqQyoEjnNrddYDNHrolQ=";
}
+ {
+ version = "10.3.0.26";
+ minCudaVersion = "11.0";
+ maxCudaVersion = "11.8";
+ cudnnVersion = "8.9";
+ filename = "TensorRT-10.3.0.26.Linux.x86_64-gnu.cuda-11.8.tar.gz";
+ hash = "sha256-1O9TwlUP0eLqTozMs53EefmjriiaHjxb4A4GIuN9jvc=";
+ }
+ {
+ version = "10.3.0.26";
+ minCudaVersion = "12.0";
+ maxCudaVersion = "12.5";
+ cudnnVersion = "8.9";
+ filename = "TensorRT-10.3.0.26.Linux.x86_64-gnu.cuda-12.5.tar.gz";
+ hash = "sha256-rf8c1avl2HATgGFyNR5Y/QJOW/D8YdSe9LhM047ZkIE=";
+ }
];
};
}