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