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