8e800cedaf
cudaPackages.cuda_compat: ignore missing libs provided at runtime cudaPackages.gpus: Jetson should never build by default cudaPackages.flags: don't build Jetson capabilities by default cudaPackages: re-introduce filter for pre-existing CUDA redist packages in overrides cudaPackages: only recurseIntoAttrs for the latest of each major version cudaPackages.nvccCompatabilities: use GCC 10 through CUDA 11.5 to avoid a GLIBC incompatability cudaPackages.cutensor: acquire libcublas through cudatoolkit prior to 11.4 cudaPackages.cuda_compat: mark as broken on aarch64-linux if not targeting Jetson cudaPackages.cutensor_1_4: fix build cudaPackages: adjust use of autoPatchelfIgnoreMissingDeps cudaPackages.cuda_nvprof: remove unecessary override to add addOpenGLRunpath cudaPackages: use getExe' to avoid patchelf warning about missing meta.mainProgram cudaPackages: fix evaluation with Nix 2.3 cudaPackages: fix platform detection for Jetson/non-Jetson aarch64-linux python3Packages.tensorrt: mark as broken if required packages are missing Note: evaluating the name of the derivation will fail if tensorrt is not present, which is why we wrap the value in `lib.optionalString`. cudaPackages.flags.getNixSystem: add guard based on jetsonTargets cudaPackages.cudnn: use explicit path to patchelf cudaPackages.tensorrt: use explicit path to patchelf
130 lines
4.5 KiB
Nix
130 lines
4.5 KiB
Nix
# NOTE: Check https://developer.nvidia.com/nvidia-tensorrt-8x-download.
|
|
# Version policy is to keep the latest minor release for each major release.
|
|
{
|
|
tensorrt.releases = {
|
|
# jetson
|
|
linux-aarch64 = [];
|
|
# powerpc
|
|
linux-ppc64le = [];
|
|
# server-grade arm
|
|
linux-sbsa = [
|
|
{
|
|
version = "8.2.5.1";
|
|
minCudaVersion = "11.4";
|
|
maxCudaVersion = "11.4";
|
|
cudnnVersion = "8.2";
|
|
filename = "TensorRT-8.2.5.1.Ubuntu-20.04.aarch64-gnu.cuda-11.4.cudnn8.2.tar.gz";
|
|
hash = "sha256-oWfQ3lq2aoMPv65THeotnMilTzP+QWqKeToLU8eO+qo=";
|
|
}
|
|
{
|
|
version = "8.4.3.1";
|
|
minCudaVersion = "11.6";
|
|
maxCudaVersion = "11.6";
|
|
cudnnVersion = "8.4";
|
|
filename = "TensorRT-8.4.3.1.Ubuntu-20.04.aarch64-gnu.cuda-11.6.cudnn8.4.tar.gz";
|
|
hash = "sha256-9tLlrB8cKYFvN2xF0Pol5CZs06iuuI5mq+6jpzD8wWI=";
|
|
}
|
|
{
|
|
version = "8.5.3.1";
|
|
minCudaVersion = "11.8";
|
|
maxCudaVersion = "11.8";
|
|
cudnnVersion = "8.6";
|
|
filename = "TensorRT-8.5.3.1.Ubuntu-20.04.aarch64-gnu.cuda-11.8.cudnn8.6.tar.gz";
|
|
hash = "sha256-GW//mX0brvN/waHo9Wd07xerOEz3X/H/HAW2ZehYtTA=";
|
|
}
|
|
{
|
|
version = "8.6.1.6";
|
|
minCudaVersion = "12.0";
|
|
maxCudaVersion = "12.0";
|
|
cudnnVersion = null;
|
|
filename = "TensorRT-8.6.1.6.Ubuntu-20.04.aarch64-gnu.cuda-12.0.tar.gz";
|
|
hash = "sha256-Lc4+v/yBr17VlecCSFMLUDlXMTYV68MGExwnUjGme5E=";
|
|
}
|
|
];
|
|
# x86_64
|
|
linux-x86_64 = [
|
|
{
|
|
version = "8.0.3.4";
|
|
minCudaVersion = "10.2";
|
|
maxCudaVersion = "10.2";
|
|
cudnnVersion = "8.2";
|
|
filename = "TensorRT-8.0.3.4.Linux.x86_64-gnu.cuda-10.2.cudnn8.2.tar.gz";
|
|
hash = "sha256-LxcXgwe1OCRfwDsEsNLIkeNsOcx3KuF5Sj+g2dY6WD0=";
|
|
}
|
|
{
|
|
version = "8.0.3.4";
|
|
minCudaVersion = "11.0";
|
|
maxCudaVersion = "11.3";
|
|
cudnnVersion = "8.2";
|
|
filename = "TensorRT-8.0.3.4.Linux.x86_64-gnu.cuda-11.3.cudnn8.2.tar.gz";
|
|
hash = "sha256-MXdDUCT/SqWm26jB7QarEcwOG/O7cS36Y6Q0IvQTE/M=";
|
|
}
|
|
{
|
|
version = "8.2.5.1";
|
|
minCudaVersion = "10.2";
|
|
maxCudaVersion = "10.2";
|
|
cudnnVersion = "8.2";
|
|
filename = "TensorRT-8.2.5.1.Linux.x86_64-gnu.cuda-10.2.cudnn8.2.tar.gz";
|
|
hash = "sha256-XV2Bf2LH8OM2GEMjV80MDweb1hSVF/wFUcaW3KP2m8Q=";
|
|
}
|
|
{
|
|
# The docs claim this supports through 11.5 despite the file name indicating 11.4.
|
|
version = "8.2.5.1";
|
|
minCudaVersion = "11.0";
|
|
maxCudaVersion = "11.5";
|
|
cudnnVersion = "8.2";
|
|
filename = "TensorRT-8.2.5.1.Linux.x86_64-gnu.cuda-11.4.cudnn8.2.tar.gz";
|
|
hash = "sha256-LcNpYvDiT7AavqzK1MRlijo2qDN7jznigeS77US713E=";
|
|
}
|
|
{
|
|
version = "8.4.3.1";
|
|
minCudaVersion = "10.2";
|
|
maxCudaVersion = "10.2";
|
|
cudnnVersion = "8.4";
|
|
filename = "TensorRT-8.4.3.1.Linux.x86_64-gnu.cuda-10.2.cudnn8.4.tar.gz";
|
|
hash = "sha256-2c3Zzt93FBWWQtrSIvpbzzS6BT9s0NzALzdwXGLOZEU=";
|
|
}
|
|
{
|
|
# The docs claim this supports through 11.7 despite the file name indicating 11.6.
|
|
version = "8.4.3.1";
|
|
minCudaVersion = "11.0";
|
|
maxCudaVersion = "11.7";
|
|
cudnnVersion = "8.4";
|
|
filename = "TensorRT-8.4.3.1.Linux.x86_64-gnu.cuda-11.6.cudnn8.4.tar.gz";
|
|
hash = "sha256-jXwghcFjncxzh1BIwjWYqFJs4wiRNoduMdkCWOSeT2E=";
|
|
}
|
|
{
|
|
version = "8.5.3.1";
|
|
minCudaVersion = "10.2";
|
|
maxCudaVersion = "10.2";
|
|
cudnnVersion = "8.6";
|
|
filename = "TensorRT-8.5.3.1.Linux.x86_64-gnu.cuda-10.2.cudnn8.6.tar.gz";
|
|
hash = "sha256-WCt6yfOmFbrjqdYCj6AE2+s2uFpISwk6urP+2I0BnGQ=";
|
|
}
|
|
{
|
|
version = "8.5.3.1";
|
|
minCudaVersion = "11.0";
|
|
maxCudaVersion = "11.8";
|
|
cudnnVersion = "8.6";
|
|
filename = "TensorRT-8.5.3.1.Linux.x86_64-gnu.cuda-11.8.cudnn8.6.tar.gz";
|
|
hash = "sha256-BNeuOYvPTUAfGxI0DVsNrX6Z/FAB28+SE0ptuGu7YDY=";
|
|
}
|
|
{
|
|
version = "8.6.1.6";
|
|
minCudaVersion = "11.0";
|
|
maxCudaVersion = "11.8";
|
|
cudnnVersion = "8.9";
|
|
filename = "TensorRT-8.6.1.6.Linux.x86_64-gnu.cuda-11.8.tar.gz";
|
|
hash = "sha256-Fb/mBT1F/uxF7McSOpEGB2sLQ/oENfJC2J3KB3gzd1k=";
|
|
}
|
|
{
|
|
version = "8.6.1.6";
|
|
minCudaVersion = "12.0";
|
|
maxCudaVersion = "12.1";
|
|
cudnnVersion = "8.9";
|
|
filename = "TensorRT-8.6.1.6.Linux.x86_64-gnu.cuda-12.0.tar.gz";
|
|
hash = "sha256-D4FXpfxTKZQ7M4uJNZE3M1CvqQyoEjnNrddYDNHrolQ=";
|
|
}
|
|
];
|
|
};
|
|
}
|