This allows cupy to find the thrust library which is needed among other things for cupy.argsort
Removed unnecessary dependencies Remove extra whitespace
This commit is contained in:
parent
019b637fb1
commit
eaeb87add1
@ -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…
Reference in New Issue
Block a user