diff --git a/pkgs/by-name/ka/katago/package.nix b/pkgs/by-name/ka/katago/package.nix index 204d6777b0c4..e3ca8cd86781 100644 --- a/pkgs/by-name/ka/katago/package.nix +++ b/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 ] diff --git a/pkgs/development/cuda-modules/tensorrt/fixup.nix b/pkgs/development/cuda-modules/tensorrt/fixup.nix index aab6d8f32079..25650d60e624 100644 --- a/pkgs/development/cuda-modules/tensorrt/fixup.nix +++ b/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. diff --git a/pkgs/development/cuda-modules/tensorrt/releases.nix b/pkgs/development/cuda-modules/tensorrt/releases.nix index c42218eebe96..13c7c1a10c32 100644 --- a/pkgs/development/cuda-modules/tensorrt/releases.nix +++ b/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="; + } ]; }; }