openrl.supports.opengpu package¶ Submodules¶ openrl.supports.opengpu.gpu_info module¶ class openrl.supports.opengpu.gpu_info.GPUInfo[source]¶ Bases: 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()[source]¶ openrl.supports.opengpu.gpu_info.get_remote_GPU_info(request_api: str)[source]¶ openrl.supports.opengpu.gpu_info.preserve_decimal(a: float, keep_num: int = 2)[source]¶ openrl.supports.opengpu.manager module¶ class openrl.supports.opengpu.manager.LocalGPUManager(args: argparse.Namespace = None)[source]¶ Bases: object get_gpu() → int[source]¶ get_learner_gpu(learner_id: int = 0) → Optional[int][source]¶ get_learner_gpus() → List[int][source]¶ get_worker_gpu(worker_id: int = 0) → int[source]¶ log_info()[source]¶ class openrl.supports.opengpu.manager.RemoteGPUManager(pytorch_config=None, check: bool = False)[source]¶ Bases: object cal_learner_number()[source]¶ check_gpus()[source]¶ get_gpu_info(server_list: list)[source]¶ Module contents¶