ilyakooo0/nixpkgs
1
0
Fork 0
mirror of https://github.com/ilyakooo0/nixpkgs.git synced 2024-11-11 15:27:20 +03:00
Code Issues Projects Releases Wiki Activity

pytorch: Move cudatoolkit to nativeBuildInputs

Browse Source 
nvcc must be available in PATH at build time; otherwise CUDA support
will be disabled.

Signed-off-by: Anders Kaseorg <andersk@mit.edu>
This commit is contained in:
Anders Kaseorg 2019-04-21 19:31:55 -07:00 committed by Frederik Rietdijk
parent a416ef58b7
commit 27d13628ee
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -81,11 +81,11 @@ in buildPythonPackage rec {
cmake
utillinux
which
];
] ++ lib.optionals cudaSupport [ cudatoolkit_joined ];
buildInputs = [
numpy.blas
] ++ lib.optionals cudaSupport [ cudatoolkit_joined cudnn ]
] ++ lib.optionals cudaSupport [ cudnn ]
++ lib.optionals stdenv.isLinux [ numactl ];
propagatedBuildInputs = [