summary refs log tree commit diff
path: root/pkgs/top-level
diff options
context:
space:
mode:
authorhyphon81 <zero812n@gmail.com>2017-12-28 17:42:53 +0100
committerJan Tojnar <jtojnar@gmail.com>2017-12-28 17:47:08 +0100
commit0184f60c1c2053f45faa017d291834db58354a6d (patch)
tree698193cf3156b32681aea3cc2aec4192d76fbaec /pkgs/top-level
parente5921012a15261d9d19edff5dfe81554db082d8f (diff)
downloadnixlib-0184f60c1c2053f45faa017d291834db58354a6d.tar
nixlib-0184f60c1c2053f45faa017d291834db58354a6d.tar.gz
nixlib-0184f60c1c2053f45faa017d291834db58354a6d.tar.bz2
nixlib-0184f60c1c2053f45faa017d291834db58354a6d.tar.lz
nixlib-0184f60c1c2053f45faa017d291834db58354a6d.tar.xz
nixlib-0184f60c1c2053f45faa017d291834db58354a6d.tar.zst
nixlib-0184f60c1c2053f45faa017d291834db58354a6d.zip
pythonPackages.cupy: init at 2.2.0
Diffstat (limited to 'pkgs/top-level')
-rw-r--r--pkgs/top-level/python-packages.nix6
1 files changed, 6 insertions, 0 deletions
diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix
index a45d8b271773..cd565923431c 100644
--- a/pkgs/top-level/python-packages.nix
+++ b/pkgs/top-level/python-packages.nix
@@ -1567,6 +1567,12 @@ in {
 
   cufflinks = callPackage ../development/python-modules/cufflinks { };
 
+  cupy = callPackage ../development/python-modules/cupy {
+    cudatoolkit = pkgs.cudatoolkit8;
+    cudnn = pkgs.cudnn6_cudatoolkit8;
+    nccl = pkgs.nccl;
+  };
+
   cx_Freeze = callPackage ../development/python-modules/cx_freeze {};
 
   cvxopt = buildPythonPackage rec {