diff options
Diffstat (limited to 'pkgs/applications/science/math/cntk/default.nix')
| -rw-r--r-- | pkgs/applications/science/math/cntk/default.nix | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/pkgs/applications/science/math/cntk/default.nix b/pkgs/applications/science/math/cntk/default.nix index e15e2a43d774..c007490c94d0 100644 --- a/pkgs/applications/science/math/cntk/default.nix +++ b/pkgs/applications/science/math/cntk/default.nix @@ -33,6 +33,9 @@ in stdenv.mkDerivation rec { # Force OpenMPI to use g++ in PATH. OMPI_CXX = "g++"; + # Uses some deprecated tensorflow functions + NIX_CFLAGS_COMPILE = "-Wno-error=deprecated-declarations"; + buildInputs = [ openblas opencv3 libzip boost protobuf mpi ] ++ lib.optional cudaSupport cudatoolkit ++ lib.optional cudnnSupport cudnn; @@ -40,7 +43,7 @@ in stdenv.mkDerivation rec { configureFlags = [ "--with-opencv=${opencv3}" "--with-libzip=${libzip.dev}" - "--with-openblas=${openblas}" + "--with-openblas=${openblas.dev}" "--with-boost=${boost.dev}" "--with-protobuf=${protobuf}" "--with-mpi=${mpi}" |