6dba41fbcb
Use the attribute mpi to provide a system wide default MPI
implementation. The default is openmpi (as before).
This now allows for overriding the MPI implentation by using
the overlay mechanism. Build all packages with mpich instead
of the default openmpi can now be achived like this:
self: super:
{
mpi = super.mpich;
}
All derivations that have been using "mpi ? null" to provide optional
building with MPI have been change in the following way to allow for
optional builds with MPI:
{ ...
, mpi
, useMpi ? false
}
62 lines
1.4 KiB
Nix
62 lines
1.4 KiB
Nix
{ lib
|
|
, buildPythonPackage
|
|
, pkgs
|
|
, numpy
|
|
, scipy
|
|
, mpi
|
|
, enum34
|
|
, protobuf
|
|
, pip
|
|
, python
|
|
, swig
|
|
}:
|
|
|
|
let
|
|
cntk = pkgs.cntk;
|
|
in
|
|
buildPythonPackage {
|
|
inherit (cntk) name version src;
|
|
|
|
nativeBuildInputs = [ swig mpi ];
|
|
buildInputs = [ cntk mpi ];
|
|
propagatedBuildInputs = [ numpy scipy enum34 protobuf pip ];
|
|
|
|
CNTK_LIB_PATH = "${cntk}/lib";
|
|
CNTK_COMPONENT_VERSION = cntk.version;
|
|
CNTK_VERSION = cntk.version;
|
|
CNTK_VERSION_BANNER = cntk.version;
|
|
|
|
postPatch = ''
|
|
cd bindings/python
|
|
sed -i 's,"libmpi.so.12","${mpi}/lib/libmpi.so",g' cntk/train/distributed.py
|
|
|
|
# Remove distro and libs checks; they aren't compatible with NixOS and besides we guarantee
|
|
# compatibility by providing a package.
|
|
cat <<EOF > cntk/cntk_py_init.py
|
|
def cntk_check_distro_info():
|
|
pass
|
|
def cntk_check_libs():
|
|
pass
|
|
EOF
|
|
'';
|
|
|
|
postInstall = ''
|
|
rm -rf $out/${python.sitePackages}/cntk/libs
|
|
ln -s ${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 (cntk.meta) homepage description license maintainers platforms;
|
|
# doesn't support Python 3.7
|
|
broken = lib.versionAtLeast python.version "3.7";
|
|
};
|
|
}
|