{ stdenv, buildPythonPackage , fetchPypi, isPy3k, linuxPackages, gcc5 , fastrlock, numpy, six, wheel, pytest, mock , cudatoolkit, cudnn, nccl }: buildPythonPackage rec { pname = "cupy"; version = "4.4.1"; src = fetchPypi { inherit pname version; sha256 = "d46a3eea628a17b9c3f446e4b13513a1f0cc0b9930214417b41c90dcf0bb427f"; }; checkInputs = [ pytest mock ]; nativeBuildInputs = [ gcc5 ]; propagatedBuildInputs = [ cudatoolkit cudnn linuxPackages.nvidia_x11 nccl fastrlock numpy six wheel ]; # In python3, test was failed... doCheck = !isPy3k; meta = with stdenv.lib; { description = "A NumPy-compatible matrix library accelerated by CUDA"; homepage = https://cupy.chainer.org/; license = licenses.mit; platforms = [ "x86_64-linux" ]; maintainers = with maintainers; [ hyphon81 ]; }; }