python37Packages.cupy: Add cutensor optional dep
This commit is contained in:
parent
a680b02816
commit
13b5995f06
|
@ -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
|
||||
|
|
|
@ -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 { };
|
||||
|
|
Loading…
Reference in a new issue