blob: cfc16063cf73c70afd92806c718b6d7176270b0b (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
|
{ stdenv
, buildPythonPackage
, pkgs
, numpy
, scipy
, enum34
, protobuf
, pip
, python
}:
buildPythonPackage rec {
inherit (pkgs.cntk) name version src;
nativeBuildInputs = [ pkgs.swig pkgs.openmpi ];
buildInputs = [ pkgs.cntk pkgs.openmpi ];
propagatedBuildInputs = [ numpy scipy enum34 protobuf pip ];
CNTK_LIB_PATH = "${pkgs.cntk}/lib";
CNTK_COMPONENT_VERSION = pkgs.cntk.version;
postPatch = ''
cd bindings/python
sed -i 's,"libmpi.so.12","${pkgs.openmpi}/lib/libmpi.so",g' cntk/train/distributed.py
'';
postInstall = ''
rm -rf $out/${python.sitePackages}/cntk/libs
ln -s ${pkgs.cntk}/lib $out/${python.sitePackages}/cntk/libs
# It's not installed for some reason.
cp cntk/cntk_py.py $out/${python.sitePackages}/cntk
'';
# Actual tests are broken.
checkPhase = ''
cd $NIX_BUILD_TOP
${python.interpreter} -c "import cntk"
'';
meta = {
inherit (pkgs.cntk.meta) homepage description license maintainers;
# broken in CNTK 2.4 starting with python-3.7
# ("ImportError: cannot import name 'GenericMeta' from 'typing'")
broken = stdenv.lib.versionAtLeast python.version "3.7";
};
}
|