tjh.dev
1{ stdenv, buildPythonPackage
2, fetchPypi, isPy3k, linuxPackages
3, fastrlock, numpy, six, wheel, pytest, mock
4, cudatoolkit, cudnn, nccl
5}:
6
7buildPythonPackage rec {
8 pname = "cupy";
9 version = "6.2.0";
10
11 src = fetchPypi {
12 inherit pname version;
13 sha256 = "0d6liaavgqks772rqam53qha3yk6dfw24i0pj3izxvvawzhlp10z";
14 };
15
16 checkInputs = [
17 pytest
18 mock
19 ];
20
21 propagatedBuildInputs = [
22 cudatoolkit
23 cudnn
24 linuxPackages.nvidia_x11
25 nccl
26 fastrlock
27 numpy
28 six
29 wheel
30 ];
31
32 # In python3, test was failed...
33 doCheck = !isPy3k;
34
35 meta = with stdenv.lib; {
36 description = "A NumPy-compatible matrix library accelerated by CUDA";
37 homepage = https://cupy.chainer.org/;
38 license = licenses.mit;
39 platforms = [ "x86_64-linux" ];
40 maintainers = with maintainers; [ hyphon81 ];
41 };
42}