cudaPackages.tensorrt: 10.3.0.26 -> 10.8.0.43 (#386143)

Author: prusnak

pkgs/by-name/ka/katago/package.nix

@@ -34,8 +34,18 @@ assert lib.assertOneOf "backend" backend [
 let
   githash = "cd0ed6c0712088ddb901be68189ba7fa1439a9e7";
   fakegit = writeShellScriptBin "git" "echo ${githash}";
+  stdenv' =
+    if
+      builtins.elem backend [
+        "cuda"
+        "tensorrt"
+      ]
+    then
+      cudaPackages.backendStdenv
+    else
+      stdenv;
 in
-stdenv.mkDerivation rec {
+stdenv'.mkDerivation rec {
   pname = "katago";
   version = "1.15.3";
 
@@ -58,10 +68,12 @@ stdenv.mkDerivation rec {
     ]
     ++ lib.optionals (backend == "eigen") [ eigen ]
     ++ lib.optionals (backend == "cuda") [
+      cudaPackages.cuda_cudart
       cudaPackages.cudnn
       cudaPackages.cudatoolkit
     ]
     ++ lib.optionals (backend == "tensorrt") [
+      cudaPackages.cuda_cudart
       cudaPackages.cudatoolkit
       cudaPackages.tensorrt
     ]

pkgs/development/cuda-modules/tensorrt/fixup.nix

@@ -75,6 +75,11 @@ finalAttrs: prevAttrs: {
         rm "$dir"
         mv "targets/${targetArch}/$dir" "$dir"
       done
+
+      # Remove broken symlinks
+      for dir in include samples; do
+        rm "targets/${targetArch}/$dir" || :
+      done
     '';
 
   # Tell autoPatchelf about runtime dependencies.

pkgs/development/cuda-modules/tensorrt/releases.nix

@@ -1,4 +1,6 @@
-# NOTE: Check https://developer.nvidia.com/nvidia-tensorrt-8x-download.
+# NOTE: Check https://developer.nvidia.com/nvidia-tensorrt-8x-download
+#             https://developer.nvidia.com/nvidia-tensorrt-10x-download
+
 # Version policy is to keep the latest minor release for each major release.
 {
   tensorrt.releases = {
@@ -109,6 +111,22 @@
         filename = "TensorRT-10.3.0.26.Linux.x86_64-gnu.cuda-12.5.tar.gz";
         hash = "sha256-rf8c1avl2HATgGFyNR5Y/QJOW/D8YdSe9LhM047ZkIE=";
       }
+      {
+        version = "10.8.0.43";
+        minCudaVersion = "11.0";
+        maxCudaVersion = "11.8";
+        cudnnVersion = "8.9";
+        filename = "TensorRT-10.8.0.43.Linux.x86_64-gnu.cuda-11.8.tar.gz";
+        hash = "sha256-ZhdJ9ZUanOSQ3TbKNEIvS+fHLQ+TXZ+SdrUL4UiER+k=";
+      }
+      {
+        version = "10.8.0.43";
+        minCudaVersion = "12.0";
+        maxCudaVersion = "12.8";
+        cudnnVersion = "9.7";
+        filename = "TensorRT-10.8.0.43.Linux.x86_64-gnu.cuda-12.8.tar.gz";
+        hash = "sha256-V31tivU4FTQUuYZ8ZmtPZYUvwusefA6jogbl+vvH1J4=";
+      }
     ];
   };
 }