pytorch: Use nativeBuildInputs to specify binary build dependencies.
This commit is contained in:
parent
80d032b13b
commit
488230c253
|
@ -77,11 +77,14 @@ in buildPythonPackage rec {
|
|||
# https://github.com/intel/mkl-dnn/commit/8134d346cdb7fe1695a2aa55771071d455fae0bc
|
||||
NIX_CFLAGS_COMPILE = lib.optionals (numpy.blasImplementation == "mkl") [ "-Wno-error=array-bounds" ];
|
||||
|
||||
buildInputs = [
|
||||
nativeBuildInputs = [
|
||||
cmake
|
||||
numpy.blas
|
||||
utillinux
|
||||
which
|
||||
];
|
||||
|
||||
buildInputs = [
|
||||
numpy.blas
|
||||
] ++ lib.optionals cudaSupport [ cudatoolkit_joined cudnn ]
|
||||
++ lib.optionals stdenv.isLinux [ numactl ];
|
||||
|
||||
|
|
Loading…
Reference in a new issue