cudaPackages.tensorrt: 10.3.0.26 -> 10.8.0.43 (#386143)
This commit is contained in:
@@ -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=";
|
||||
}
|
||||
];
|
||||
};
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user