code: def total_memory(self, device_index=None):
      return torch.cuda.get_device_properties(device_index).total_memory
  
imports: from deepspeed.accelerator.abstract_accelerator import DeepSpeedAccelerator
  import torch.cuda
label_desc: Use image node_id 33 for calling the CUDA_Accelerator obj's underlying member method code with example usage: obj.total_memory(device_index) and returns: torch