tensorrt: provide default version fallback
When passed a CUDA version not in the table of default TensorRT versions for CUDA versions, fall back to a default package version instead of breaking evaluation of the package.
This commit is contained in:
parent
35425566b8
commit
bcf0c3a882
|
@ -58,6 +58,6 @@ final: prev: let
|
|||
"11.4" = "8.4.0";
|
||||
"11.5" = "8.4.0";
|
||||
"11.6" = "8.4.0";
|
||||
}.${cudaVersion};
|
||||
}.${cudaVersion} or "8.4.0";
|
||||
|
||||
in tensorRTPackages
|
||||
|
|
Loading…
Reference in a new issue