python3Packages.jaxlib-bin: comment explaining cudatoolkit dependency

This commit is contained in:
Samuel Ainsworth 2022-02-18 21:27:28 +00:00
parent a632c843a7
commit ab9d59200c

View File

@ -97,7 +97,8 @@ buildPythonPackage rec {
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;
pythonImportsCheck = [ "jaxlib" ];