From 605ce577fad5bbc661b58d2f3ca5881f04a6a749 Mon Sep 17 00:00:00 2001 From: Jonathan Ringer Date: Thu, 2 Jun 2022 09:12:13 -0700 Subject: [PATCH] python3Packages.pytorch: add pybind11 to buildInputs This allows for the headers to be available, so that pytorch doesn't try and install pybind11 into its own includes directory. --- pkgs/development/python-modules/pytorch/default.nix | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/pkgs/development/python-modules/pytorch/default.nix b/pkgs/development/python-modules/pytorch/default.nix index d6e78c31d753..299b4eba0cf0 100644 --- a/pkgs/development/python-modules/pytorch/default.nix +++ b/pkgs/development/python-modules/pytorch/default.nix @@ -176,6 +176,10 @@ in buildPythonPackage rec { USE_MKLDNN = setBool mklDnnSupport; USE_MKLDNN_CBLAS = setBool mklDnnSupport; + # Avoid using pybind11 from git submodule + # Also avoids pytorch exporting the headers of pybind11 + USE_SYSTEM_BIND11 = true; + preBuild = '' export MAX_JOBS=$NIX_BUILD_CORES ${python.interpreter} setup.py build --cmake-only @@ -223,7 +227,7 @@ in buildPythonPackage rec { removeReferencesTo ] ++ lib.optionals cudaSupport [ cudatoolkit_joined ]; - buildInputs = [ blas blas.provider ] + buildInputs = [ blas blas.provider pybind11 ] ++ lib.optionals cudaSupport [ cudnn magma nccl ] ++ lib.optionals stdenv.isLinux [ numactl ];