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

Removed unnecessary dependencies

Remove extra whitespace
wip/yesman
Christoph Stich 5 years ago committed by Frederik Rietdijk
parent 019b637fb1
commit eaeb87add1
  1. 9
      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/;

Loading…
Cancel
Save