diff --git a/pkgs/applications/science/math/mxnet/default.nix b/pkgs/applications/science/math/mxnet/default.nix new file mode 100644 index 00000000000..67183c9dd55 --- /dev/null +++ b/pkgs/applications/science/math/mxnet/default.nix @@ -0,0 +1,43 @@ +{ stdenv, lib, fetchgit, cmake +, opencv, gtest, openblas, liblapack +, cudaSupport ? false, cudatoolkit +, cudnnSupport ? false, cudnn +}: + +assert cudnnSupport -> cudaSupport; + +stdenv.mkDerivation rec { + name = "mxnet-${version}"; + version = "0.11.0"; + + # Submodules needed + src = fetchgit { + url = "https://github.com/apache/incubator-mxnet"; + rev = "refs/tags/${version}"; + sha256 = "1qgns0c70a1gfyil96h17ms736nwdkp9kv496gvs9pkzqzvr6cpz"; + }; + + nativeBuildInputs = [ cmake ]; + + buildInputs = [ opencv gtest openblas liblapack ] + ++ lib.optional cudaSupport cudatoolkit + ++ lib.optional cudnnSupport cudnn; + + cmakeFlags = lib.optional (!cudaSupport) "-DUSE_CUDA=OFF" + ++ lib.optional (!cudnnSupport) "-DUSE_CUDNN=OFF"; + + installPhase = '' + install -Dm755 libmxnet.so $out/lib/libmxnet.so + cp -r ../include $out + ''; + + enableParallelBuilding = true; + + meta = with stdenv.lib; { + description = "Lightweight, Portable, Flexible Distributed/Mobile Deep Learning with Dynamic, Mutation-aware Dataflow Dep Scheduler"; + homepage = "https://mxnet.incubator.apache.org/"; + maintainers = with maintainers; [ abbradar ]; + license = licenses.asl20; + platforms = platforms.linux; + }; +} diff --git a/pkgs/top-level/all-packages.nix b/pkgs/top-level/all-packages.nix index 61c35ca93b1..3662baec608 100644 --- a/pkgs/top-level/all-packages.nix +++ b/pkgs/top-level/all-packages.nix @@ -18472,6 +18472,8 @@ with pkgs; sbcl = null; }; + mxnet = callPackage ../applications/science/math/mxnet { }; + wxmaxima = callPackage ../applications/science/math/wxmaxima { wxGTK = wxGTK30; }; pari = callPackage ../applications/science/math/pari { tex = texlive.combined.scheme-basic; };