diff options
author | Frederik Rietdijk <freddyrietdijk@fridh.nl> | 2016-06-21 14:26:51 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2016-06-21 14:26:51 +0200 |
commit | 8ef747baed3a3313c229d86e8c410b731d3a9124 (patch) | |
tree | f86c3b494439d5e8473a40fa2288950bd42e0244 /pkgs/top-level | |
parent | 0a8e613cea27d4a32ea43e73c2b5bf0ebcb54211 (diff) | |
parent | 465d6a211370ed635fac338b59e718b4056f1142 (diff) | |
download | nixlib-8ef747baed3a3313c229d86e8c410b731d3a9124.tar nixlib-8ef747baed3a3313c229d86e8c410b731d3a9124.tar.gz nixlib-8ef747baed3a3313c229d86e8c410b731d3a9124.tar.bz2 nixlib-8ef747baed3a3313c229d86e8c410b731d3a9124.tar.lz nixlib-8ef747baed3a3313c229d86e8c410b731d3a9124.tar.xz nixlib-8ef747baed3a3313c229d86e8c410b731d3a9124.tar.zst nixlib-8ef747baed3a3313c229d86e8c410b731d3a9124.zip |
Merge pull request #16334 from artuuge/theano_cudnn
Theano-cuda: init at 0.8.2
Diffstat (limited to 'pkgs/top-level')
-rw-r--r-- | pkgs/top-level/all-packages.nix | 4 | ||||
-rw-r--r-- | pkgs/top-level/python-packages.nix | 15 |
2 files changed, 19 insertions, 0 deletions
diff --git a/pkgs/top-level/all-packages.nix b/pkgs/top-level/all-packages.nix index 5b4b6936cc13..60beb8eae28d 100644 --- a/pkgs/top-level/all-packages.nix +++ b/pkgs/top-level/all-packages.nix @@ -1194,6 +1194,10 @@ in cudnn = callPackage ../development/libraries/science/math/cudnn/default.nix {}; + cudnn5_cudatoolkit75 = callPackage ../development/libraries/science/math/cudnn/7.5-5.0 { + cudatoolkit = self.cudatoolkit75; + }; + curlFull = self.curl.override { idnSupport = true; ldapSupport = true; diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index cd7982baa503..8ba9e066b439 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -21031,6 +21031,21 @@ in modules // { }; }; + Theano-cuda = callPackage ../development/python-modules/theano/cuda ( + let + boost = pkgs.boost159.override { + inherit (self) python numpy scipy; + }; + in rec { + cudatoolkit = pkgs.cudatoolkit75; + cudnn = pkgs.cudnn5_cudatoolkit75; + inherit (self) numpy scipy; + pycuda = self.pycuda.override { inherit boost; }; + libgpuarray = self.libgpuarray-cuda.override { + clblas = pkgs.clblas-cuda.override { inherit boost; }; + }; + }); + tidylib = buildPythonPackage rec { version = "0.2.4"; name = "pytidylib-${version}"; |