U 'DgN@sUdZddlZddlZddlmZddlmZzddlZdZWne k rZdZdZYnXddl m Z m Z m Z mZddlmZdd lmZdd lmZeeZd d ged gdZeed<dZdZdZedddZeeeeddddZ dS)z: invalid choice: 'install'zTemplate: linux/nvidia/latelink Type: boolean Default: true Description: Late-link NVIDIA kernel modules? Enable this to link the NVIDIA kernel modules in cloud-init and make them available for use. X_LOADTEMPLATEFILE)distroc Cst|ts tdt||fd}tt||}|sLt d||dSt dsjt d|dgd}t|d}|r|d |7}t d |||r|nd t j|d d }tj|d}zpz2t|ttd} | t|W5QRXWn8tk r0} zttdt| W5d} ~ XYnXW5tj|rNt|Xzt ddd|gWnVt jk r} z4t| j krt!dnd| j"krt!dW5d} ~ XYnXdS)Nz.'drivers' config expected dict, found '%s': %sznvidia/license-acceptedz$Not installing NVIDIA drivers. %s=%szubuntu-driverszI'ubuntu-drivers' command not available. Installing ubuntu-drivers-commonzubuntu-drivers-commonZnvidiaznvidia/versionz:{}zs4         @