pyrox.dev
mxnet: 0.11.0 -> 1.1.0
+7
-4
pkgs/applications/science/math/mxnet/default.nix
+7
-4
pkgs/applications/science/math/mxnet/default.nix
···
8
9
stdenv.mkDerivation rec {
10
name = "mxnet-${version}";
11
-
version = "0.11.0";
12
13
# Submodules needed
14
src = fetchgit {
···
23
++ lib.optionals cudaSupport [ cudatoolkit nvidia_x11 ]
24
++ lib.optional cudnnSupport cudnn;
25
26
-
cmakeFlags = [
27
-
(if cudaSupport then "-DCUDA_ARCH_NAME=All" else "-DUSE_CUDA=OFF")
28
-
] ++ lib.optional (!cudnnSupport) "-DUSE_CUDNN=OFF";
29
30
installPhase = ''
31
install -Dm755 libmxnet.so $out/lib/libmxnet.so
···
8
9
stdenv.mkDerivation rec {
10
name = "mxnet-${version}";
11
+
version = "1.1.0";
12
13
# Submodules needed
14
src = fetchgit {
···
23
++ lib.optionals cudaSupport [ cudatoolkit nvidia_x11 ]
24
++ lib.optional cudnnSupport cudnn;
25
26
+
cmakeFlags =
27
+
(if cudaSupport then [
28
+
"-DCUDA_ARCH_NAME=All"
29
+
"-DCUDA_HOST_COMPILER=${cudatoolkit.cc}/bin/cc"
30
+
] else [ "-DUSE_CUDA=OFF" ])
31
+
++ lib.optional (!cudnnSupport) "-DUSE_CUDNN=OFF";
32
33
installPhase = ''
34
install -Dm755 libmxnet.so $out/lib/libmxnet.so