From a13f5d0da1549479e65927b37d7897dce037a2f2 Mon Sep 17 00:00:00 2001 From: Julien Dehos Date: Fri, 12 Apr 2019 12:31:22 +0200 Subject: mxnet: fix cuda support enabled build (#48781) --- pkgs/applications/science/math/mxnet/default.nix | 1 + 1 file changed, 1 insertion(+) (limited to 'pkgs/applications') diff --git a/pkgs/applications/science/math/mxnet/default.nix b/pkgs/applications/science/math/mxnet/default.nix index 061f48bb3d93b..9c1a2e2028237 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" ]) -- cgit 1.4.1