mxnet: 1.2.1 -> 1.4.1

This commit is contained in:
Nikolay Amiantov 2019-06-02 20:11:58 +03:00
parent c534d5c865
commit 1deb406ac2
2 changed files with 7 additions and 9 deletions

View file

@ -1,5 +1,5 @@
{ config, stdenv, lib, fetchurl, bash, cmake
, opencv, gtest, openblas, liblapack, perl
, opencv3, gtest, openblas, liblapack, perl
, cudaSupport ? config.cudaSupport or false, cudatoolkit, nvidia_x11
, cudnnSupport ? cudaSupport, cudnn
}:
@ -8,24 +8,22 @@ assert cudnnSupport -> cudaSupport;
stdenv.mkDerivation rec {
name = "mxnet-${version}";
version = "1.2.1";
version = "1.4.1";
# Fetching from git does not work at the time (1.2.1) due to an
# incorrect hash in one of the submodules. The provided tarballs
# contain all necessary sources.
src = fetchurl {
url = "https://github.com/apache/incubator-mxnet/releases/download/${version}/apache-mxnet-src-${version}-incubating.tar.gz";
sha256 = "053zbdgs4j8l79ipdz461zc7wyfbfcflmi5bw7lj2q08zm1glnb2";
sha256 = "1d0lhlpdaxycjzpwwrpgjd3v2q2ka89v5rr13ddxayy7ld2hxiaj";
};
nativeBuildInputs = [ cmake perl ];
buildInputs = [ opencv gtest openblas liblapack ]
buildInputs = [ opencv3 gtest openblas liblapack ]
++ lib.optionals cudaSupport [ cudatoolkit nvidia_x11 ]
++ lib.optional cudnnSupport cudnn;
cmakeFlags =
(if cudaSupport then [
[ "-DUSE_MKL_IF_AVAILABLE=OFF" ]
++ (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"

View file

@ -21,7 +21,7 @@ buildPythonPackage rec {
postPatch = ''
substituteInPlace python/setup.py \
--replace "graphviz<0.9.0," "graphviz" \
--replace "numpy<=1.15.0," "numpy" \
--replace "numpy<=1.15.2," "numpy" \
--replace "requests<2.19.0," "requests"
'';