cntk: move defaults to package file

This commit is contained in:
Jan Malakhovski 2019-02-03 15:32:16 +00:00
parent 31aa884b7f
commit 4bc66b1994
2 changed files with 5 additions and 8 deletions

View File

@ -1,8 +1,8 @@
{ lib, stdenv, fetchgit, fetchFromGitHub, cmake { config, lib, stdenv, fetchgit, fetchFromGitHub, cmake
, openblas, opencv3, libzip, boost, protobuf, openmpi , openblas, opencv3, libzip, boost, protobuf, openmpi
, onebitSGDSupport ? false , onebitSGDSupport ? false
, cudaSupport ? false, cudatoolkit, nvidia_x11 , cudaSupport ? config.cudaSupport or false, cudatoolkit, nvidia_x11
, cudnnSupport ? false, cudnn , cudnnSupport ? cudaSupport, cudnn
}: }:
assert cudnnSupport -> cudaSupport; assert cudnnSupport -> cudaSupport;

View File

@ -21862,12 +21862,9 @@ in
opencv3 = opencv3WithoutCuda; # Used only for image loading. opencv3 = opencv3WithoutCuda; # Used only for image loading.
}); });
cntk = callPackage ../applications/science/math/cntk rec { cntk = callPackage ../applications/science/math/cntk {
cudaSupport = pkgs.config.cudaSupport or false;
cudnnSupport = cudaSupport;
inherit (linuxPackages) nvidia_x11; inherit (linuxPackages) nvidia_x11;
# Used only for image loading. opencv3 = opencv3WithoutCuda; # Used only for image loading.
opencv3 = opencv3WithoutCuda;
}; };
ecm = callPackage ../applications/science/math/ecm { }; ecm = callPackage ../applications/science/math/ecm { };