nordic-dev.net/nixpkgs
nordic-dev.net/nixpkgs
2
0
Fork 0
mirror of https://github.com/NixOS/nixpkgs.git synced 2024-12-08 14:53:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

tensorrt: fix for cudaVerison 11.7

Browse Source
This commit is contained in:
David Guibert 2022-08-07 12:13:18 +02:00
parent df956cf4c1
commit feccefa684
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
pkgs/development/libraries/science/math/tensorrt/extension.nix
View File

@ -35,7 +35,7 @@ final: prev: let
fullVersion = "8.4.0.6";
sha256 = "sha256-DNgHHXF/G4cK2nnOWImrPXAkOcNW6Wy+8j0LRpAH/LQ=";
tarball = "TensorRT-${fullVersion}.Linux.x86_64-gnu.cuda-${fileVersionCuda}.cudnn${fileVersionCudnn}.tar.gz";
supportedCudaVersions = [ "11.0" "11.1" "11.2" "11.3" "11.4" "11.5" "11.6" ];
supportedCudaVersions = [ "11.0" "11.1" "11.2" "11.3" "11.4" "11.5" "11.6" "11.7" ];
}
rec {
fileVersionCuda = "10.2";
@ -58,6 +58,7 @@ final: prev: let
"11.4" = "8.4.0";
"11.5" = "8.4.0";
"11.6" = "8.4.0";
"11.7" = "8.4.0";
}.${cudaVersion} or "8.4.0";
in tensorRTPackages