diff --git a/pkgs/development/python-modules/onnxruntime/default.nix b/pkgs/development/python-modules/onnxruntime/default.nix index 6d52ed8ca512..250bea6c2290 100644 --- a/pkgs/development/python-modules/onnxruntime/default.nix +++ b/pkgs/development/python-modules/onnxruntime/default.nix @@ -10,6 +10,7 @@ re2, # dependencies + openvino, coloredlogs, numpy, packaging, @@ -76,6 +77,7 @@ buildPythonPackage { ++ lib.optionals onnxruntime.passthru.ncclSupport [ nccl # libnccl.so.XX ] + ++ lib.optionals stdenv.hostPlatform.isLinux [ openvino ] ); dependencies = [ diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 7e4d095655c1..dd384f1b3452 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -11835,6 +11835,7 @@ self: super: with self; { python3Packages = self; pythonSupport = true; }; + inherit (pkgs) openvino; }; onnxruntime-tools = callPackage ../development/python-modules/onnxruntime-tools { };