Unverified Commit 5a140d8d authored by Pavol Rusnak's avatar Pavol Rusnak Committed by GitHub
Browse files

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

parents c91461ae 496278b6

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

+13 −1
Original line number Diff line number Diff line
@@ -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

+5 −0
Original line number Diff line number Diff line
@@ -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

+19 −1
Original line number Diff line number Diff line 
# 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=";

      }

    ];

  };

}