[no-relnote] Use nbody as test job

Signed-off-by: Evan Lezar <[email protected]>
NVIDIA · Aug 28, 2024 · cc174b1 · cc174b1
1 parent 337ac67
commit cc174b1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/e2e/common/gpu_job.go b/tests/e2e/common/gpu_job.go
@@ -38,7 +38,8 @@ var GPUJob = &batchv1.Job{
 				Containers: []v1.Container{
 					{
 						Name:  "cuda-container",
-						Image: "nvcr.io/nvidia/k8s/cuda-sample:vectoradd-cuda10.2",
+						Image: "nvcr.io/nvidia/k8s/cuda-sample:nbody-cuda11.7.1-ubuntu18.04",
+						Args:  []string{"--benchmark", "--numbodies=10000"},
 						Resources: v1.ResourceRequirements{
 							Limits: v1.ResourceList{
 								"nvidia.com/gpu": resource.MustParse("1"),