Theano-cuda: init at 0.8.2
This commit is contained in:
62
pkgs/development/python-modules/theano/cuda/default.nix
Normal file
62
pkgs/development/python-modules/theano/cuda/default.nix
Normal file
@@ -0,0 +1,62 @@
|
||||
{ buildPythonPackage
|
||||
, fetchFromGitHub
|
||||
, blas
|
||||
, numpy
|
||||
, six
|
||||
, scipy
|
||||
, nose
|
||||
, nose-parameterized
|
||||
, pydot_ng
|
||||
, sphinx
|
||||
, pygments
|
||||
, libgpuarray
|
||||
, python
|
||||
, pycuda
|
||||
, cudatoolkit
|
||||
, cudnn
|
||||
, stdenv
|
||||
}:
|
||||
|
||||
buildPythonPackage rec {
|
||||
name = "Theano-cuda-${version}";
|
||||
version = "0.8.2";
|
||||
|
||||
src = fetchFromGitHub {
|
||||
owner = "Theano";
|
||||
repo = "Theano";
|
||||
rev = "46fbfeb628220b5e42bf8277a5955c52d153e874";
|
||||
sha256 = "1sl91gli3jaw5gpjqqab4fiq4x6282spqciaid1s65pjsf3k55sc";
|
||||
};
|
||||
|
||||
doCheck = false;
|
||||
|
||||
patchPhase = ''
|
||||
pushd theano/sandbox/gpuarray
|
||||
sed -i -re '2s/^/from builtins import bytes\n/g' subtensor.py
|
||||
sed -i -re "s/(b'2')/int(bytes(\1))/g" subtensor.py
|
||||
sed -i -re "s/(ctx.bin_id\[\-2\])/int(\1)/g" subtensor.py
|
||||
|
||||
sed -i -re '2s/^/from builtins import bytes\n/g' dnn.py
|
||||
sed -i -re "s/(b'30')/int(bytes(\1))/g" dnn.py
|
||||
sed -i -re "s/(ctx.bin_id\[\-2:\])/int(\1)/g" dnn.py
|
||||
popd
|
||||
'';
|
||||
|
||||
dontStrip = true;
|
||||
|
||||
propagatedBuildInputs = [
|
||||
blas
|
||||
numpy
|
||||
six
|
||||
scipy
|
||||
nose
|
||||
nose-parameterized
|
||||
pydot_ng
|
||||
sphinx
|
||||
pygments
|
||||
pycuda
|
||||
cudatoolkit
|
||||
libgpuarray
|
||||
] ++ (stdenv.lib.optional (cudnn != null) [ cudnn ]);
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user