diff options
author | hyphon81 <zero812n@gmail.com> | 2017-12-28 17:42:53 +0100 |
---|---|---|
committer | Jan Tojnar <jtojnar@gmail.com> | 2017-12-28 17:47:08 +0100 |
commit | 0184f60c1c2053f45faa017d291834db58354a6d (patch) | |
tree | 698193cf3156b32681aea3cc2aec4192d76fbaec /pkgs/development/python-modules/cupy | |
parent | e5921012a15261d9d19edff5dfe81554db082d8f (diff) | |
download | nixlib-0184f60c1c2053f45faa017d291834db58354a6d.tar nixlib-0184f60c1c2053f45faa017d291834db58354a6d.tar.gz nixlib-0184f60c1c2053f45faa017d291834db58354a6d.tar.bz2 nixlib-0184f60c1c2053f45faa017d291834db58354a6d.tar.lz nixlib-0184f60c1c2053f45faa017d291834db58354a6d.tar.xz nixlib-0184f60c1c2053f45faa017d291834db58354a6d.tar.zst nixlib-0184f60c1c2053f45faa017d291834db58354a6d.zip |
pythonPackages.cupy: init at 2.2.0
Diffstat (limited to 'pkgs/development/python-modules/cupy')
-rw-r--r-- | pkgs/development/python-modules/cupy/default.nix | 46 |
1 files changed, 46 insertions, 0 deletions
diff --git a/pkgs/development/python-modules/cupy/default.nix b/pkgs/development/python-modules/cupy/default.nix new file mode 100644 index 000000000000..6ac91c0aa255 --- /dev/null +++ b/pkgs/development/python-modules/cupy/default.nix @@ -0,0 +1,46 @@ +{ stdenv, python, buildPythonPackage +, fetchPypi, isPy3k, linuxPackages, gcc5 +, fastrlock, numpy, six, wheel, pytest, mock +, cudatoolkit, cudnn, nccl +}: + +buildPythonPackage rec { + pname = "cupy"; + version = "2.2.0"; + + src = fetchPypi { + inherit pname version; + sha256 = "0si0ri8azxvxh3lpm4l4g60jf4nwzibi53yldbdbzb1svlqq060r"; + }; + + 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 ]; + }; +} |