diff options
author | John Ericson <John.Ericson@Obsidian.Systems> | 2021-03-15 23:09:15 +0000 |
---|---|---|
committer | John Ericson <John.Ericson@Obsidian.Systems> | 2021-03-17 20:15:51 +0000 |
commit | 13b5995f06ec0ff5efeefd7ca6667d2acf5fb254 (patch) | |
tree | 8b9fc177b81dbde997333148b8c9dca39b50180a /pkgs/development/python-modules/cupy | |
parent | a680b02816260761e6ed609bdfc6fecb250e5784 (diff) | |
download | nixpkgs-13b5995f06ec0ff5efeefd7ca6667d2acf5fb254.tar nixpkgs-13b5995f06ec0ff5efeefd7ca6667d2acf5fb254.tar.gz nixpkgs-13b5995f06ec0ff5efeefd7ca6667d2acf5fb254.tar.bz2 nixpkgs-13b5995f06ec0ff5efeefd7ca6667d2acf5fb254.tar.lz nixpkgs-13b5995f06ec0ff5efeefd7ca6667d2acf5fb254.tar.xz nixpkgs-13b5995f06ec0ff5efeefd7ca6667d2acf5fb254.tar.zst nixpkgs-13b5995f06ec0ff5efeefd7ca6667d2acf5fb254.zip |
python37Packages.cupy: Add cutensor optional dep
Diffstat (limited to 'pkgs/development/python-modules/cupy')
-rw-r--r-- | pkgs/development/python-modules/cupy/default.nix | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/pkgs/development/python-modules/cupy/default.nix b/pkgs/development/python-modules/cupy/default.nix index 5c4978a388f..d4b42ac83e8 100644 --- a/pkgs/development/python-modules/cupy/default.nix +++ b/pkgs/development/python-modules/cupy/default.nix @@ -1,7 +1,7 @@ { lib, buildPythonPackage , fetchPypi, isPy3k, linuxPackages , fastrlock, numpy, six, wheel, pytest, mock, setuptools -, cudatoolkit, cudnn, nccl +, cudatoolkit, cudnn, cutensor, nccl }: buildPythonPackage rec { @@ -26,6 +26,7 @@ buildPythonPackage rec { propagatedBuildInputs = [ cudatoolkit cudnn + cutensor linuxPackages.nvidia_x11 nccl fastrlock |