Merge pull request #931 from elezar/update-gpu-job

[no-relnote] Use nbody as test job
NVIDIA · Aug 28, 2024 · dd73702 · dd73702
2 parents 337ac67 + cc174b1
commit dd73702
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"),