code: def get_cuda_mem():
      torch.cuda.synchronize()
      gc.collect()
      return torch.cuda.memory_allocated()
  
file_path: fairscale/tests/nn/data_parallel/test_fsdp_summon_full_params.py
imports: import functools
  import gc
  import unittest
  from parameterized import parameterized
  import pytest
  import torch
  from fairscale.internal.version import torch_version
  from .test_fsdp import CONFIG_OPTIONS, DistributedTest, rename_test, spawn_and_init
label_desc: Use image node_id 1 for calling a global function with example usage: get_cuda_mem() and returns: torch