diff options
Diffstat (limited to 'pkgs/development')
-rw-r--r-- | pkgs/development/python-modules/cupy/default.nix | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/pkgs/development/python-modules/cupy/default.nix b/pkgs/development/python-modules/cupy/default.nix index 5426a609449d..443d96b097a1 100644 --- a/pkgs/development/python-modules/cupy/default.nix +++ b/pkgs/development/python-modules/cupy/default.nix @@ -1,6 +1,6 @@ { stdenv, buildPythonPackage , fetchPypi, isPy3k, linuxPackages -, fastrlock, numpy, six, wheel, pytest, mock +, fastrlock, numpy, six, wheel, pytest, mock, setuptools , cudatoolkit, cudnn, nccl }: @@ -18,6 +18,10 @@ buildPythonPackage rec { mock ]; + preConfigure = '' + export CUDA_PATH=${cudatoolkit} + ''; + propagatedBuildInputs = [ cudatoolkit cudnn @@ -26,12 +30,15 @@ buildPythonPackage rec { fastrlock numpy six + setuptools wheel ]; # In python3, test was failed... doCheck = !isPy3k; + enableParallelBuilding = true; + meta = with stdenv.lib; { description = "A NumPy-compatible matrix library accelerated by CUDA"; homepage = https://cupy.chainer.org/; |