class_docstr: Utility Context Manager to start a multi GPU context using dask_cuda
  Parameters:
  -----------
  number_of_devices : int
      Number of devices to use, verification must be done prior to call to
      ensure that there are enough devices available. If not specified, the
      cluster will be initialized to use all visible devices.
  rmm_managed_memory : bool
      True to enable managed memory (UVM) in RMM as part of the
      cluster. Default is False.
  p2p : bool
      Initialize UCX endpoints if True. Default is False.
class_docstr_tok: ["Utility","Context","Manager","to","start","a","multi","GPU","context","using","dask_cuda","Parameters",":","--","--","--","--","--","-","number_of_devices",":","int","Number","of","devices","to","use",",","verification","must","be","done","prior","to","call","to","ensure","that","there","are","enough","devices","available",".","If","not","specified",",","the","cluster","will","be","initialized","to","use","all","visible","devices",".","rmm_managed_memory",":","bool","True","to","enable","managed","memory","(","UVM",")","in","RMM","as","part","of","the","cluster",".","Default","is","False",".","p2p",":","bool","Initialize","UCX","endpoints","if","True",".","Default","is","False","."]
code: def __init__(
      self, number_of_devices=None, rmm_managed_memory=False, p2p=False
  ):
      self._number_of_devices = number_of_devices
      self._rmm_managed_memory = rmm_managed_memory
      self._client = None
      self._p2p = p2p
      self._cluster = CUDACluster(
          n_workers=self._number_of_devices,
          rmm_managed_memory=self._rmm_managed_memory,
      )
  
file_path: cugraph/python/cugraph/cugraph/tests/utils/mg_context.py
imports: import time
  import pytest
  import cugraph.dask.comms
  from dask.distributed import Client
  from dask_cuda import LocalCUDACluster
  from cugraph.dask.common.mg_utils import get_visible_devices
label_desc: Use image node_id 1 to create a new MGContext object  with example: obj = MGContext(number_of_devices, rmm_managed_memory, p2p)