jax.print_environment_info

jax.print_environment_info(return_string=False)

返回一个包含本地环境和 JAX 安装信息的字符串。

在提问或提交 bug 时，包含这些信息非常有用。

参数： return_string (bool) ：如果为 True，则返回字符串而不是打印到标准输出。

参数:

return_string (bool)

返回类型:

str | None