commit c71d8a1aeb87cf703faf3f72fd228a038a9c9723 · pyrox.dev/nixpkgs

+43

pkgs/applications/science/math/mxnet/default.nix

···

       1
       1
       +
       { stdenv, lib, fetchgit, cmake

     

       2
       2
       +
       , opencv, gtest, openblas, liblapack

     

       3
       3
       +
       , cudaSupport ? false, cudatoolkit

     

       4
       4
       +
       , cudnnSupport ? false, cudnn

     

       5
       5
       +
       }:

     

       6
       6
       +
       

     

       7
       7
       +
       assert cudnnSupport -> cudaSupport;

     

       8
       8
       +
       

     

       9
       9
       +
       stdenv.mkDerivation rec {

     

       10
       10
       +
         name = "mxnet-${version}";

     

       11
       11
       +
         version = "0.11.0";

     

       12
       12
       +
       

     

       13
       13
       +
         # Submodules needed

     

       14
       14
       +
         src = fetchgit {

     

       15
       15
       +
           url = "https://github.com/apache/incubator-mxnet";

     

       16
       16
       +
           rev = "refs/tags/${version}";

     

       17
       17
       +
           sha256 = "1qgns0c70a1gfyil96h17ms736nwdkp9kv496gvs9pkzqzvr6cpz";

     

       18
       18
       +
         };

     

       19
       19
       +
       

     

       20
       20
       +
         nativeBuildInputs = [ cmake ];

     

       21
       21
       +
       

     

       22
       22
       +
         buildInputs = [ opencv gtest openblas liblapack ]

     

       23
       23
       +
                     ++ lib.optional cudaSupport cudatoolkit

     

       24
       24
       +
                     ++ lib.optional cudnnSupport cudnn;

     

       25
       25
       +
       

     

       26
       26
       +
         cmakeFlags = lib.optional (!cudaSupport) "-DUSE_CUDA=OFF"

     

       27
       27
       +
                   ++ lib.optional (!cudnnSupport) "-DUSE_CUDNN=OFF";

     

       28
       28
       +
       

     

       29
       29
       +
         installPhase = ''

     

       30
       30
       +
           install -Dm755 libmxnet.so $out/lib/libmxnet.so

     

       31
       31
       +
           cp -r ../include $out

     

       32
       32
       +
         '';

     

       33
       33
       +
       

     

       34
       34
       +
         enableParallelBuilding = true;

     

       35
       35
       +
       

     

       36
       36
       +
         meta = with stdenv.lib; {

     

       37
       37
       +
           description = "Lightweight, Portable, Flexible Distributed/Mobile Deep Learning with Dynamic, Mutation-aware Dataflow Dep Scheduler";

     

       38
       38
       +
           homepage = "https://mxnet.incubator.apache.org/";

     

       39
       39
       +
           maintainers = with maintainers; [ abbradar ];

     

       40
       40
       +
           license = licenses.asl20;

     

       41
       41
       +
           platforms = platforms.linux;

     

       42
       42
       +
         };

     

       43
       43
       +
       }

pkgs/top-level/all-packages.nix

···

       18472
       18472
        
           sbcl = null;

     

       18473
       18473
        
         };

     

       18474
       18474
        
       

     

       18475
       18475
       +
         mxnet = callPackage ../applications/science/math/mxnet { };

     

       18476
       18476
       +
       

     

       18475
       18477
        
         wxmaxima = callPackage ../applications/science/math/wxmaxima { wxGTK = wxGTK30; };

     

       18476
       18478
        
       

     

       18477
       18479
        
         pari = callPackage ../applications/science/math/pari { tex = texlive.combined.scheme-basic; };