python3Packages.jaxlib-bin: comment explaining cudatoolkit dependency
This commit is contained in:
parent
a632c843a7
commit
ab9d59200c
@ -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" ];
|
||||
|
Loading…
Reference in New Issue
Block a user