Adding NVML init and shutdown code