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