pyrox.dev / nixpkgs
lol
fork atom

python37Packages.cupy: Add cutensor optional dep

John Ericson 13b5995f a680b028

+3 -1
unified split
+2 -1
pkgs/development/python-modules/cupy/default.nix
··· 1 1 { lib, buildPythonPackage 2 2 , fetchPypi, isPy3k, linuxPackages 3 3 , fastrlock, numpy, six, wheel, pytest, mock, setuptools 4 - , cudatoolkit, cudnn, nccl 4 + , cudatoolkit, cudnn, cutensor, nccl 5 5 }: 6 6 7 7 buildPythonPackage rec { ··· 26 26 propagatedBuildInputs = [ 27 27 cudatoolkit 28 28 cudnn 29 + cutensor 29 30 linuxPackages.nvidia_x11 30 31 nccl 31 32 fastrlock
+1
pkgs/top-level/python-packages.nix
··· 1589 1589 cudatoolkit = pkgs.cudatoolkit_10_0; 1590 1590 cudnn = pkgs.cudnn_cudatoolkit_10_0; 1591 1591 nccl = pkgs.nccl_cudatoolkit_10; 1592 + cutensor = pkgs.cutensor_cudatoolkit_10; 1592 1593 }; 1593 1594 1594 1595 curio = callPackage ../development/python-modules/curio { };