pyrox.dev
lol
This allows cupy to find the thrust library which is needed among other things for cupy.argsort
Removed unnecessary dependencies
Remove extra whitespace
authored by
Christoph Stich
and committed by
Frederik Rietdijk
eaeb87ad
019b637f
+8
-1
+8
-1
pkgs/development/python-modules/cupy/default.nix
+8
-1
pkgs/development/python-modules/cupy/default.nix
···
1
1
{ stdenv, buildPythonPackage
2
2
, fetchPypi, isPy3k, linuxPackages
3
-
, fastrlock, numpy, six, wheel, pytest, mock
3
+
, fastrlock, numpy, six, wheel, pytest, mock, setuptools
4
4
, cudatoolkit, cudnn, nccl
5
5
}:
6
6
···
18
18
mock
19
19
];
20
20
21
+
preConfigure = ''
22
+
export CUDA_PATH=${cudatoolkit}
23
+
'';
24
+
21
25
propagatedBuildInputs = [
22
26
cudatoolkit
23
27
cudnn
···
26
30
fastrlock
27
31
numpy
28
32
six
33
+
setuptools
29
34
wheel
30
35
];
31
36
32
37
# In python3, test was failed...
33
38
doCheck = !isPy3k;
39
+
40
+
enableParallelBuilding = true;
34
41
35
42
meta = with stdenv.lib; {
36
43
description = "A NumPy-compatible matrix library accelerated by CUDA";