forked from mirrors/nixpkgs
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
|
||||
, onebitSGDSupport ? false
|
||||
, cudaSupport ? false, cudatoolkit, nvidia_x11
|
||||
, cudnnSupport ? false, cudnn
|
||||
, cudaSupport ? config.cudaSupport or false, cudatoolkit, nvidia_x11
|
||||
, cudnnSupport ? cudaSupport, cudnn
|
||||
}:
|
||||
|
||||
assert cudnnSupport -> cudaSupport;
|
||||
|
|
|
@ -21862,12 +21862,9 @@ in
|
|||
opencv3 = opencv3WithoutCuda; # Used only for image loading.
|
||||
});
|
||||
|
||||
cntk = callPackage ../applications/science/math/cntk rec {
|
||||
cudaSupport = pkgs.config.cudaSupport or false;
|
||||
cudnnSupport = cudaSupport;
|
||||
cntk = callPackage ../applications/science/math/cntk {
|
||||
inherit (linuxPackages) nvidia_x11;
|
||||
# Used only for image loading.
|
||||
opencv3 = opencv3WithoutCuda;
|
||||
opencv3 = opencv3WithoutCuda; # Used only for image loading.
|
||||
};
|
||||
|
||||
ecm = callPackage ../applications/science/math/ecm { };
|
||||
|
|
Loading…
Reference in a new issue