alioth/nixpkgs
3
0
Fork 0
forked from mirrors/nixpkgs
Code Releases Activity

mxnet: 0.11.0 -> 1.1.0

Browse source
This commit is contained in:
Nikolay Amiantov 2018-02-25 20:02:20 +03:00
parent 6f96122d5b
commit 3179002f61
1 changed files with 7 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
pkgs/applications/science/math/mxnet/default.nix
View file

@ -8,7 +8,7 @@ assert cudnnSupport -> cudaSupport;
stdenv.mkDerivation rec {
name = "mxnet-${version}";
version = "0.11.0";
version = "1.1.0";
# Submodules needed
src = fetchgit {
@ -23,9 +23,12 @@ stdenv.mkDerivation rec {
++ lib.optionals cudaSupport [ cudatoolkit nvidia_x11 ]
++ lib.optional cudnnSupport cudnn;
cmakeFlags = [
(if cudaSupport then "-DCUDA_ARCH_NAME=All" else "-DUSE_CUDA=OFF")
] ++ lib.optional (!cudnnSupport) "-DUSE_CUDNN=OFF";
cmakeFlags =
(if cudaSupport then [
"-DCUDA_ARCH_NAME=All"
"-DCUDA_HOST_COMPILER=${cudatoolkit.cc}/bin/cc"
] else [ "-DUSE_CUDA=OFF" ])
++ lib.optional (!cudnnSupport) "-DUSE_CUDNN=OFF";
installPhase = ''
install -Dm755 libmxnet.so $out/lib/libmxnet.so