diff options
author | artuuge <artuuge@gmail.com> | 2016-06-19 10:52:21 +0200 |
---|---|---|
committer | artuuge <artuuge@gmail.com> | 2016-06-20 01:06:07 +0200 |
commit | 465d6a211370ed635fac338b59e718b4056f1142 (patch) | |
tree | 6b20f79c4d2b36e13817110deffd727eb5edd513 /pkgs/development/python-modules | |
parent | 0243ccfbffaffa8cac669f5b80c28ee4bb5334e6 (diff) | |
download | nixlib-465d6a211370ed635fac338b59e718b4056f1142.tar nixlib-465d6a211370ed635fac338b59e718b4056f1142.tar.gz nixlib-465d6a211370ed635fac338b59e718b4056f1142.tar.bz2 nixlib-465d6a211370ed635fac338b59e718b4056f1142.tar.lz nixlib-465d6a211370ed635fac338b59e718b4056f1142.tar.xz nixlib-465d6a211370ed635fac338b59e718b4056f1142.tar.zst nixlib-465d6a211370ed635fac338b59e718b4056f1142.zip |
Theano-cuda: init at 0.8.2
Diffstat (limited to 'pkgs/development/python-modules')
-rw-r--r-- | pkgs/development/python-modules/theano/cuda/default.nix | 62 |
1 files changed, 62 insertions, 0 deletions
diff --git a/pkgs/development/python-modules/theano/cuda/default.nix b/pkgs/development/python-modules/theano/cuda/default.nix new file mode 100644 index 000000000000..bf49d3918611 --- /dev/null +++ b/pkgs/development/python-modules/theano/cuda/default.nix @@ -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 ]); + +} |