code: def OnGPU(gpu_id):
      """A utility function that returns a device option protobuf of the
      specified gpu id.
      """
      device_option = caffe2_pb2.DeviceOption()
      device_option.device_type = workspace.GpuDeviceType
      device_option.device_id = gpu_id
      return device_option
  
file_path: pytorch/caffe2/python/muji.py
imports: import numpy
  from caffe2.proto import caffe2_pb2
  from caffe2.python import workspace
label_desc: Use image node_id 1 for calling a global function with example usage: OnGPU(gpu_id) and returns: device_option