ray.rllib.utils.framework.尝试导入_tf#

ray.rllib.utils.framework.try_import_tf(error: bool = False)[源代码]#

尝试导入 tf 并返回模块(或 None)。

参数:

error – 如果无法导入tf,是否引发错误。

返回:

包含以下内容的元组:1) tf1.x 模块(来自 tf2.x.compat.v1 或 tf1.x)。2) tf 模块(来自 import tensorflow)。可以是 tf1.x 或 2.x。3) 实际安装的 tf 版本,以整数表示:1 或 2。

抛出:

ImportError – 如果 error=True 且未安装 tf。