talexander/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Releases Activity

python3Packages.onnxruntime: buildInputs dep on onnxruntime

Browse Source 
The onnxruntime libs are patched to look for additional runtime
libraries in `${onnxruntime}/lib`, so we must depend on it.

Alternatively, onnxruntime could be built twice, once for each lib path
patch, but onnxruntime takes a vert long time to build with cuda support.
This commit is contained in:
Mitchell Pleune 2025-02-17 19:17:52 +00:00
parent 9d3ae807eb
commit eb4009247f
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
pkgs/development/python-modules/onnxruntime/default.nix
View File

@ -50,6 +50,13 @@ buildPythonPackage {
oneDNN oneDNN
re2 re2
onnxruntime.protobuf onnxruntime.protobuf
# https://github.com/NixOS/nixpkgs/pull/357656 patches the onnx lib to ${pkgs.onnxruntime}/lib
# but these files are copied into this package too. If the origional non-python onnxruntime
# package is GC-ed, cuda support in this python package will break.
# Two options, rebuild onnxruntime twice with the different paths hard-coded, or just hold a runtime
# dependency between the two. Option 2, because onnxruntime takes forever to build with cuda support.
onnxruntime
] ]
++ lib.optionals onnxruntime.passthru.cudaSupport ( ++ lib.optionals onnxruntime.passthru.cudaSupport (
with onnxruntime.passthru.cudaPackages; with onnxruntime.passthru.cudaPackages;