ray.rllib.utils.framework.尝试导入_torch

ray.rllib.utils.framework.try_import_torch(error: bool = False)

尝试导入 torch 并返回模块（或 None）。

参数:

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

返回:

由 torch 和 torch.nn 模块组成的元组。

抛出:

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