def init_nvidia_config()

in src/worker/exporters/custom_exporter.py [0:0]

• 18 lines of code
• 3 McCabe index (conditional complexity)

def init_nvidia_config():
    """Initialize NVIDIA configuration"""
    global config
    global GPU_Mapping
    # Check if it's an NVIDIA VM
    nvArch = os.path.exists('/dev/nvidiactl')
    if nvArch:
        cmd = 'nvidia-smi -L'
        result = shell_cmd(cmd, 5)
        gpuCount = len(result.split('\nGPU'))
        try:
            for gpu in range(gpuCount):
                cmd = f'nvidia-smi -q -g {gpu} -d ACCOUNTING'
                result = shell_cmd(cmd, 5)
                pci = re.search(r"\w+:\w\w:\w\w\.", result).group().lower()
                pci = pci.replace('.', '')[-10:]
                GPU_Mapping[pci] = str(gpu)  # PCI mapping
                config['gpu_id'][str(gpu)] = GPU_Mapping[pci]
        except Exception as e:
            logging.error(f"Error initializing NVIDIA config: {e}")