fix(hypervisor): add DEVICE_PLUGIN_PATH env to hypervisor container (#454)

Author: 0x5457

internal/constants/env.go

@@ -145,6 +145,7 @@ const (
 	HypervisorMetricsFormatEnv      = "TF_HYPERVISOR_METRICS_FORMAT"
 	HypervisorMetricsExtraLabelsEnv = "TF_HYPERVISOR_METRICS_EXTRA_LABELS"
 	HypervisorDetectUsedGPUEnv      = "DETECT_IN_USED_GPUS"
+	HypervisorDevicePluginPathEnv   = "DEVICE_PLUGIN_PATH"
 
 	// Add ptrace capability to hypervisor container, to trace all host PID using GPU
 	SystemPtraceCapability = "SYS_PTRACE"

internal/utils/compose.go

@@ -641,6 +641,11 @@ func composeHypervisorContainer(spec *v1.PodSpec, pool *tfv1.GPUPool, enableVect
 		spec.Containers[0].Image = pool.Spec.ComponentConfig.Hypervisor.Image
 	}
 
+	spec.Containers[0].Env = append(spec.Containers[0].Env, v1.EnvVar{
+		Name:  constants.HypervisorDevicePluginPathEnv,
+		Value: constants.KubeletDevicePluginPath,
+	})
+
 	if len(spec.Containers[0].Resources.Requests) == 0 {
 		spec.Containers[0].Resources.Requests = hypervisorDefaultRequests
 	}