From a13f5d0da1549479e65927b37d7897dce037a2f2 Mon Sep 17 00:00:00 2001 From: Julien Dehos Date: Fri, 12 Apr 2019 12:31:22 +0200 Subject: [PATCH] mxnet: fix cuda support enabled build (#48781) --- pkgs/applications/science/math/mxnet/default.nix | 1 + 1 file changed, 1 insertion(+) diff --git a/pkgs/applications/science/math/mxnet/default.nix b/pkgs/applications/science/math/mxnet/default.nix index 061f48bb3d9..9c1a2e20282 100644 --- a/pkgs/applications/science/math/mxnet/default.nix +++ b/pkgs/applications/science/math/mxnet/default.nix @@ -26,6 +26,7 @@ stdenv.mkDerivation rec { cmakeFlags = (if cudaSupport then [ + "-DUSE_OLDCMAKECUDA=ON" # see https://github.com/apache/incubator-mxnet/issues/10743 "-DCUDA_ARCH_NAME=All" "-DCUDA_HOST_COMPILER=${cudatoolkit.cc}/bin/cc" ] else [ "-DUSE_CUDA=OFF" ])