tjh.dev
1{ stdenv, lib
2, buildPythonPackage, fetchPypi, isPy3k
3, filelock, protobuf, numpy, pytest, mock
4, cupy, cudaSupport ? false
5}:
6
7buildPythonPackage rec {
8 pname = "chainer";
9 version = "5.2.0";
10
11 src = fetchPypi {
12 inherit pname version;
13 sha256 = "cc8390a7f445a14a1bc71d54de348be247158fe2813a5ef11c5046265001c8c4";
14 };
15
16 checkInputs = [
17 pytest
18 mock
19 ];
20
21 propagatedBuildInputs = [
22 filelock
23 protobuf
24 numpy
25 ] ++ lib.optionals cudaSupport [ cupy ];
26
27 # In python3, test was failed...
28 doCheck = !isPy3k;
29
30 meta = with stdenv.lib; {
31 description = "A flexible framework of neural networks for deep learning";
32 homepage = https://chainer.org/;
33 license = licenses.mit;
34 maintainers = with maintainers; [ hyphon81 ];
35 };
36}