code: def _print_memory_used():
      allocated = torch.cuda.max_memory_allocated() / 2**30
      reserved = torch.cuda.max_memory_reserved() / 2**30
      print(f"Allocated: {allocated:.2f} GB")
      print(f"Reserved: {reserved:.2f} GB")
  
file_path: vissl/tests/bench_vit_fsdp.py
imports: import argparse
  import contextlib
  from dataclasses import dataclass
  import torch
  import torch.multiprocessing
  import torch.nn
  from vissl.models import build_model
  from vissl.utils.fsdp_utils import fsdp_wrapper
  from vissl.utils.hydra_config import compose_hydra_configuration, convert_to_attrdict
  from vissl.utils.layer_memory_tracking import LayerwiseMemoryTracker
  from vissl.utils.profiler import create_runtime_profiler
  from vissl.utils.test_utils import in_temporary_directory, init_distributed_on_file, with_temp_files, with_timing
label_desc: Use image node_id 2 for calling the BenchVitFSDP obj's underlying member method code with example usage: obj._print_memory_used() without return types