diff --git a/default.nix b/default.nix index 64ba35d93..59a4ceaaa 100644 --- a/default.nix +++ b/default.nix @@ -1,6 +1,8 @@ { lib , buildPythonPackage , pythonRelaxDepsHook +, pybind11 +, boost , pytorchWithCuda11 , torchvisionWithCuda11 , torchWithoutCuda @@ -41,6 +43,11 @@ buildPythonPackage rec { pythonRelaxDepsHook ]; + buildInputs = [ + pybind11 + boost + ]; + pythonRelaxDeps = [ "matplotlib" "torchvision"