Merge pull request #160772 from samuela/samuela/jaxlib-bin

python3Packages.jaxlib-bin: comment explaining cudatoolkit dependency
This commit is contained in:
Samuel Ainsworth
2022-02-19 11:55:00 -08:00
committed by GitHub
@@ -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" ];