The Python bindings to onnxruntime were added by me in #193188. Adding Python support this way is not a good way. Here a wheel was created (which is fine) and installed in a python output. The propagated-build-inputs file is put in the dev output. That's fine, except that stdenv.mkDerivation does not automatically add dev when python is included, because other outputs are only added when no output is explicitly selected. This means that when you want to use these bindings in another Python package, pip will complain it cannot find the dependencies of the bindings. In this PR, the onnxruntime derivation outputs a wheel in the dist output. Then, in python-packages.nix we have a separate onnxruntime package which installs the bindings. The Python bindings have quite some dependencies which, depending on your use case, are not required. Thus the dependency relax hook is used to remove some of these dependencies. Note there is also an issue with protobuf versions. The onnxruntime bindings require an older protobuf and Python protobuf which we cannot offer. Thus protobuf is also removed as Python dependency.
65 lines
1.4 KiB
Nix
65 lines
1.4 KiB
Nix
{ lib
|
|
, buildPythonPackage
|
|
, autoPatchelfHook
|
|
, pythonRelaxDepsHook
|
|
, onnxruntime
|
|
, coloredlogs
|
|
, numpy
|
|
, packaging
|
|
, oneDNN
|
|
|
|
}:
|
|
|
|
# onnxruntime requires an older protobuf.
|
|
# Doing an override in protobuf in the python-packages set
|
|
# can give you a functioning Python package but note not
|
|
# all Python packages will be compatible then.
|
|
#
|
|
# Because protobuf is not always needed we remove it
|
|
# as a runtime dependency from our wheel.
|
|
#
|
|
# We do include here the non-Python protobuf so the shared libs
|
|
# link correctly. If you do also want to include the Python
|
|
# protobuf, you can add it to your Python env, but be aware
|
|
# the version likely mismatches with what is used here.
|
|
|
|
buildPythonPackage {
|
|
inherit (onnxruntime) pname version;
|
|
format = "wheel";
|
|
src = onnxruntime.dist;
|
|
|
|
unpackPhase = ''
|
|
cp -r $src dist
|
|
chmod +w dist
|
|
'';
|
|
|
|
nativeBuildInputs = [
|
|
autoPatchelfHook
|
|
pythonRelaxDepsHook
|
|
];
|
|
|
|
# This project requires fairly large dependencies such as sympy which we really don't always need.
|
|
pythonRemoveDeps = [
|
|
"flatbuffers"
|
|
"protobuf"
|
|
"sympy"
|
|
];
|
|
|
|
# Libraries are not linked correctly.
|
|
buildInputs = [
|
|
oneDNN
|
|
onnxruntime.protobuf
|
|
];
|
|
|
|
propagatedBuildInputs = [
|
|
coloredlogs
|
|
# flatbuffers
|
|
numpy
|
|
packaging
|
|
# protobuf
|
|
# sympy
|
|
];
|
|
|
|
meta = onnxruntime.meta // { maintainers = with lib.maintainers; [ fridh ]; };
|
|
}
|