Reduce memory requirement for pytorch job

opendatahub-io · Dec 4, 2024 · 03d5ee3 · 03d5ee3
1 parent a3ff979
commit 03d5ee3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/kfto/core/kfto_training_test.go b/tests/kfto/core/kfto_training_test.go
@@ -164,12 +164,12 @@ func createKFTOPyTorchJob(test Test, namespace string, config corev1.ConfigMap,
 									Resources: corev1.ResourceRequirements{
 										Requests: corev1.ResourceList{
 											corev1.ResourceCPU:            resource.MustParse("2"),
-											corev1.ResourceMemory:         resource.MustParse("16Gi"),
+											corev1.ResourceMemory:         resource.MustParse("8Gi"),
 											corev1.ResourceName(gpuLabel): resource.MustParse(fmt.Sprint(numGpus)),
 										},
 										Limits: corev1.ResourceList{
 											corev1.ResourceCPU:            resource.MustParse("2"),
-											corev1.ResourceMemory:         resource.MustParse("16Gi"),
+											corev1.ResourceMemory:         resource.MustParse("8Gi"),
 											corev1.ResourceName(gpuLabel): resource.MustParse(fmt.Sprint(numGpus)),
 										},
 									},