diff options
Diffstat (limited to 'pkgs/development/python-modules/onnxruntime/default.nix')
-rw-r--r-- | pkgs/development/python-modules/onnxruntime/default.nix | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/pkgs/development/python-modules/onnxruntime/default.nix b/pkgs/development/python-modules/onnxruntime/default.nix index 612e76354a00e..48c342c1626f7 100644 --- a/pkgs/development/python-modules/onnxruntime/default.nix +++ b/pkgs/development/python-modules/onnxruntime/default.nix @@ -3,7 +3,6 @@ stdenv, buildPythonPackage, autoPatchelfHook, - pythonRelaxDepsHook, onnxruntime, coloredlogs, numpy, @@ -36,7 +35,7 @@ buildPythonPackage { chmod +w dist ''; - nativeBuildInputs = [ pythonRelaxDepsHook ] ++ lib.optionals stdenv.isLinux [ autoPatchelfHook ]; + nativeBuildInputs = lib.optionals stdenv.hostPlatform.isLinux [ autoPatchelfHook ]; # This project requires fairly large dependencies such as sympy which we really don't always need. pythonRemoveDeps = [ @@ -60,6 +59,7 @@ buildPythonPackage { libcufft # libcufft.so.XX cudnn # libcudnn.soXX cuda_cudart # libcudart.so.XX + nccl # libnccl.so.XX ] ); |