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

This commit is contained in:
Pavol Rusnak
2025-03-01 17:56:48 +01:00
committed by GitHub
3 changed files with 37 additions and 2 deletions
+13 -1
View File
@@ -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
]
@@ -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.
@@ -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=";
}
];
};
}