nathan/nixpkgs
1
0
Fork 0
mirror of https://github.com/NixOS/nixpkgs.git synced 2026-08-25 17:55:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #116666 from obsidiansystems/cutensor

Browse Source 
cudatensor: Init at 1.2.2
This commit is contained in:
John Ericson
2021-03-18 13:08:16 -04:00
committed by GitHub
parent 29ad1e7737 66447439a0
commit 417658beae
9 changed files with 150 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
pkgs/development/python-modules/cupy/default.nix
View File

@@ -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