JakeHillion/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

python3Packages.pytorch: fix build

Browse Source
This commit is contained in:
Jonathan Ringer 2019-09-28 18:49:37 -07:00 committed by Frederik Rietdijk
parent b3b123a104
commit 02648a6cc1
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
pkgs/development/python-modules/pytorch/default.nix
View File

@ -44,6 +44,7 @@ in buildPythonPackage rec {
})
];
dontUseCmakeConfigure = true;
preConfigure = lib.optionalString cudaSupport ''
export CC=${cudatoolkit.cc}/bin/gcc CXX=${cudatoolkit.cc}/bin/g++
'' + lib.optionalString (cudaSupport && cudnn != null) ''