code: def probe_gpu_variant(
      mxnet_features: Dict[str, bool]
  ) -> Optional[str]:
      """
      Returns the mxnet gpu variant depending on which mxnet features are enabled
      :param mxnet_features: An mxnet feature dictionary of feature to boolean (True = enabled)
      :return: The mxnet gpu variant, eg. cu102, cu102mkl, etc.
      :raises RuntimeError is the CUDA feature is not enabled in the library
      """
      if not mxnet_features["CUDA"]:
          raise RuntimeError(
              "Cannot determine gpu variant. CUDA feature is disabled."
          )
  
      cuda_version = get_cuda_version()
      if cuda_version:
          variant = "cu{}".format(cuda_version)
          if not mxnet_features["ONEDNN"]:
              RuntimeError(
                  "Error determining mxnet variant: oneDNN should be enabled for cuda variants"
              )
          logger.debug("variant is: {}".format(variant))
          return variant
  
      raise RuntimeError(
          "Error determining mxnet variant: Could not retrieve cuda version"
      )
  
imports: import argparse
  import ctypes
  import glob
  import logging
  import os
  import re
  import sys
  from itertools import chain
  from subprocess import CalledProcessError, check_output
  from typing import Dict, List, Optional
  import boto3
  import botocore.exceptions
  import yaml
label_desc: Use image node_id 13 for calling a global function with example usage: probe_gpu_variant(mxnet_features) and returns: variant