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 | |
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')
-rw-r--r-- | pkgs/development/python-modules/cupy/default.nix | 3 | ||||
-rw-r--r-- | pkgs/top-level/python-packages.nix | 1 |
2 files changed, 3 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 diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 1f3b5e682ba..67dee805518 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -1589,6 +1589,7 @@ in { cudatoolkit = pkgs.cudatoolkit_10_0; cudnn = pkgs.cudnn_cudatoolkit_10_0; nccl = pkgs.nccl_cudatoolkit_10; + cutensor = pkgs.cutensor_cudatoolkit_10; }; curio = callPackage ../development/python-modules/curio { }; |