movefasta/nixpkgs
0
0
Fork 0
mirror of https://github.com/NixOS/nixpkgs.git synced 2025-07-13 21:50:33 +03:00
Code Activity

python3Packages.jaxlib-bin: comment explaining cudatoolkit dependency

Browse source
This commit is contained in:
Samuel Ainsworth 2022-02-18 21:27:28 +00:00
parent a632c843a7
commit ab9d59200c
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
pkgs/development/python-modules/jaxlib/bin.nix
View file

@ -97,7 +97,8 @@ buildPythonPackage rec {
done done
''; '';
# pip dependencies and optionally cudatoolkit. # pip dependencies and optionally cudatoolkit. Note that cudatoolkit is
# necessary since jaxlib looks for "ptxas" in $PATH.
propagatedBuildInputs = [ absl-py flatbuffers scipy ] ++ lib.optional cudaSupport cudatoolkit_11; propagatedBuildInputs = [ absl-py flatbuffers scipy ] ++ lib.optional cudaSupport cudatoolkit_11;
pythonImportsCheck = [ "jaxlib" ]; pythonImportsCheck = [ "jaxlib" ];