diff --git a/test/e2e/scheduling/nvidia-gpus.go b/test/e2e/scheduling/nvidia-gpus.go index 37d50f7672..01bbaa93f6 100644 --- a/test/e2e/scheduling/nvidia-gpus.go +++ b/test/e2e/scheduling/nvidia-gpus.go @@ -54,7 +54,7 @@ func makeCudaAdditionDevicePluginTestPod() *v1.Pod { RestartPolicy: v1.RestartPolicyNever, Containers: []v1.Container{ { - Name: "vector-addition", + Name: "vector-addition-cuda8", Image: imageutils.GetE2EImage(imageutils.CudaVectorAdd), Resources: v1.ResourceRequirements{ Limits: v1.ResourceList{ @@ -62,6 +62,15 @@ func makeCudaAdditionDevicePluginTestPod() *v1.Pod { }, }, }, + { + Name: "vector-addition-cuda10", + Image: imageutils.GetE2EImage(imageutils.CudaVectorAdd2), + Resources: v1.ResourceRequirements{ + Limits: v1.ResourceList{ + gpuResourceName: *resource.NewQuantity(1, resource.DecimalSI), + }, + }, + }, }, }, }