diff options
author | Julien Dehos <dehos@univ-littoral.fr> | 2019-04-12 12:31:22 +0200 |
---|---|---|
committer | Joachim F <joachifm@users.noreply.github.com> | 2019-04-12 10:31:22 +0000 |
commit | a13f5d0da1549479e65927b37d7897dce037a2f2 (patch) | |
tree | 8b433105400571341f9280cb2b92661af18caed2 /pkgs/applications/science/math/mxnet | |
parent | 728c6b75829ec1bbb45a33db128973768feb3046 (diff) |
mxnet: fix cuda support enabled build (#48781)
Diffstat (limited to 'pkgs/applications/science/math/mxnet')
-rw-r--r-- | pkgs/applications/science/math/mxnet/default.nix | 1 |
1 files changed, 1 insertions, 0 deletions
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" ]) |