openrl.supports.opengpu package¶ Submodules¶ openrl.supports.opengpu.gpu_info module¶ class openrl.supports.opengpu.gpu_info.GPUInfo[源代码]¶ 基类：object gpu_id: int¶ memory_free: int¶ memory_total: int¶ memory_used: int¶ product_name: str¶ real_id: int¶ openrl.supports.opengpu.gpu_info.get_local_GPU_info()[源代码]¶ openrl.supports.opengpu.gpu_info.get_remote_GPU_info(request_api: str)[源代码]¶ openrl.supports.opengpu.gpu_info.preserve_decimal(a: float, keep_num: int = 2)[源代码]¶ openrl.supports.opengpu.manager module¶ class openrl.supports.opengpu.manager.LocalGPUManager(args: argparse.Namespace = None)[源代码]¶ 基类：object get_gpu() → int[源代码]¶ get_learner_gpu(learner_id: int = 0) → Optional[int][源代码]¶ get_learner_gpus() → List[int][源代码]¶ get_worker_gpu(worker_id: int = 0) → int[源代码]¶ log_info()[源代码]¶ class openrl.supports.opengpu.manager.RemoteGPUManager(pytorch_config=None, check: bool = False)[源代码]¶ 基类：object cal_learner_number()[源代码]¶ check_gpus()[源代码]¶ get_gpu_info(server_list: list)[源代码]¶ Module contents¶