forked from mirrors/nixpkgs
8413a541d5
Original was a mix of config changes and code changes with a search and replace that also changed unintended bits such as messages.
84 lines
2.6 KiB
Nix
84 lines
2.6 KiB
Nix
{ stdenv
|
|
, lib
|
|
, fetchPypi
|
|
, gcc
|
|
, writeScriptBin
|
|
, buildPythonPackage
|
|
, isPyPy
|
|
, pythonOlder
|
|
, isPy3k
|
|
, nose
|
|
, numpy
|
|
, scipy
|
|
, six
|
|
, libgpuarray
|
|
, cudaSupport ? false, cudatoolkit
|
|
, cudnnSupport ? false, cudnn
|
|
, nvidia_x11
|
|
}:
|
|
|
|
assert cudnnSupport -> cudaSupport;
|
|
|
|
assert cudaSupport -> nvidia_x11 != null
|
|
&& cudatoolkit != null
|
|
&& cudnn != null;
|
|
|
|
let
|
|
extraFlags =
|
|
lib.optionals cudaSupport [ "-I ${cudatoolkit}/include" "-L ${cudatoolkit}/lib" ]
|
|
++ lib.optionals cudnnSupport [ "-I ${cudnn}/include" "-L ${cudnn}/lib" ]
|
|
++ lib.optionals cudaSupport [ "-I ${libgpuarray}/include" "-L ${libgpuarray}/lib" ];
|
|
|
|
gcc_ = writeScriptBin "g++" ''
|
|
#!${stdenv.shell}
|
|
export NIX_CC_WRAPPER_${stdenv.cc.infixSalt}_TARGET_HOST=1
|
|
export NIX_CFLAGS_COMPILE="$NIX_CFLAGS_COMPILE ${toString extraFlags}"
|
|
exec ${gcc}/bin/g++ "$@"
|
|
'';
|
|
|
|
libgpuarray_ = libgpuarray.override { inherit cudaSupport cudatoolkit; };
|
|
|
|
in buildPythonPackage rec {
|
|
pname = "Theano";
|
|
version = "1.0.2";
|
|
|
|
disabled = isPyPy || pythonOlder "2.6" || (isPy3k && pythonOlder "3.3");
|
|
|
|
src = fetchPypi {
|
|
inherit pname version;
|
|
sha256 = "6768e003d328a17011e6fca9126fbb8a6ffd3bb13cb21c450f3e724cca29abde";
|
|
};
|
|
|
|
postPatch = ''
|
|
substituteInPlace theano/configdefaults.py \
|
|
--replace 'StrParam(param, is_valid=warn_cxx)' 'StrParam('\'''${gcc_}/bin/g++'\''', is_valid=warn_cxx)' \
|
|
--replace 'rc == 0 and config.cxx != ""' 'config.cxx != ""'
|
|
'' + stdenv.lib.optionalString cudaSupport ''
|
|
substituteInPlace theano/configdefaults.py \
|
|
--replace 'StrParam(get_cuda_root)' 'StrParam('\'''${cudatoolkit}'\''')'
|
|
'' + stdenv.lib.optionalString cudnnSupport ''
|
|
substituteInPlace theano/configdefaults.py \
|
|
--replace 'StrParam(default_dnn_base_path)' 'StrParam('\'''${cudnn}'\''')'
|
|
'';
|
|
|
|
preCheck = ''
|
|
mkdir -p check-phase
|
|
export HOME=$(pwd)/check-phase
|
|
'';
|
|
doCheck = false;
|
|
# takes far too long, also throws "TypeError: sort() missing 1 required positional argument: 'a'"
|
|
# when run from the installer, and testing with Python 3.5 hits github.com/Theano/Theano/issues/4276,
|
|
# the fix for which hasn't been merged yet.
|
|
|
|
# keep Nose around since running the tests by hand is possible from Python or bash
|
|
checkInputs = [ nose ];
|
|
propagatedBuildInputs = [ numpy numpy.blas scipy six libgpuarray_ ];
|
|
|
|
meta = with stdenv.lib; {
|
|
homepage = http://deeplearning.net/software/theano/;
|
|
description = "A Python library for large-scale array computation";
|
|
license = licenses.bsd3;
|
|
maintainers = with maintainers; [ maintainers.bcdarwin ];
|
|
};
|
|
}
|