public/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

This allows cupy to find the thrust library which is needed among other things for cupy.argsort

Browse Source 
Removed unnecessary dependencies

Remove extra whitespace
This commit is contained in:
Christoph Stich 2020-01-14 12:26:09 +00:00 committed by Frederik Rietdijk
parent 019b637fb1
commit eaeb87add1
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
pkgs/development/python-modules/cupy/default.nix
View File

@ -1,6 +1,6 @@
{ stdenv, buildPythonPackage { stdenv, buildPythonPackage
, fetchPypi, isPy3k, linuxPackages , fetchPypi, isPy3k, linuxPackages
, fastrlock, numpy, six, wheel, pytest, mock , fastrlock, numpy, six, wheel, pytest, mock, setuptools
, cudatoolkit, cudnn, nccl , cudatoolkit, cudnn, nccl
}: }:
@ -18,6 +18,10 @@ buildPythonPackage rec {
mock mock
]; ];
preConfigure = ''
export CUDA_PATH=${cudatoolkit}
'';
propagatedBuildInputs = [ propagatedBuildInputs = [
cudatoolkit cudatoolkit
cudnn cudnn
@ -26,12 +30,15 @@ buildPythonPackage rec {
fastrlock fastrlock
numpy numpy
six six
setuptools
wheel wheel
]; ];
# In python3, test was failed... # In python3, test was failed...
doCheck = !isPy3k; doCheck = !isPy3k;
enableParallelBuilding = true;
meta = with stdenv.lib; { meta = with stdenv.lib; {
description = "A NumPy-compatible matrix library accelerated by CUDA"; description = "A NumPy-compatible matrix library accelerated by CUDA";
homepage = https://cupy.chainer.org/; homepage = https://cupy.chainer.org/;