diff options
Diffstat (limited to 'pkgs/development/python-modules/tensorrt/default.nix')
-rw-r--r-- | pkgs/development/python-modules/tensorrt/default.nix | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/pkgs/development/python-modules/tensorrt/default.nix b/pkgs/development/python-modules/tensorrt/default.nix index 475e7627e4baa..e108b1a773cba 100644 --- a/pkgs/development/python-modules/tensorrt/default.nix +++ b/pkgs/development/python-modules/tensorrt/default.nix @@ -11,7 +11,7 @@ let in buildPythonPackage rec { pname = "tensorrt"; - version = cudaPackages.tensorrt.version; + version = lib.optionalString (cudaPackages ? tensorrt) cudaPackages.tensorrt.version; src = cudaPackages.tensorrt.src; @@ -48,5 +48,8 @@ buildPythonPackage rec { license = licenses.unfree; platforms = [ "x86_64-linux" ]; maintainers = with maintainers; [ aidalgol ]; + broken = + !(cudaPackages ? tensorrt) + || !(cudaPackages ? cudnn); }; } |