cntk: move defaults to package file
This commit is contained in:
parent
31aa884b7f
commit
4bc66b1994
@ -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;
|
||||||
|
@ -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 { };
|
||||||
|
Loading…
x
Reference in New Issue
Block a user