code: def _init():
      torch.cuda.set_device(0)
      torch.manual_seed(0)
      np.random.seed(0)
      device = torch.device("cuda")
      offload_device = torch.device("cpu")
      return device, offload_device
  
file_path: fairscale/tests/experimental/nn/test_offload.py
imports: import contextlib
  import copy
  import numpy
  import pytest
  import torch
  from fairscale.experimental.nn.offload import OffloadModel
  from fairscale.fair_dev.testing.testing import skip_if_no_cuda
  from fairscale.internal import torch_version
label_desc: Use image node_id 1 for calling a global function with example usage: _init() and returns: device,  offload_device