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

python3Packages.pytorch: fix build

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 '' preConfigure = lib.optionalString cudaSupport ''
export CC=${cudatoolkit.cc}/bin/gcc CXX=${cudatoolkit.cc}/bin/g++ export CC=${cudatoolkit.cc}/bin/gcc CXX=${cudatoolkit.cc}/bin/g++
'' + lib.optionalString (cudaSupport && cudnn != null) '' '' + lib.optionalString (cudaSupport && cudnn != null) ''