diff --git a/test/e2e/mnist_pip_requirements.txt b/test/e2e/mnist_pip_requirements.txt index 87edeef2..4c9d5fcb 100644 --- a/test/e2e/mnist_pip_requirements.txt +++ b/test/e2e/mnist_pip_requirements.txt @@ -1,3 +1,3 @@ -pytorch_lightning==1.5.10 +pytorch_lightning==1.9.5 torchmetrics==0.9.1 torchvision==0.12.0 diff --git a/test/e2e/mnist_rayjob_raycluster_test.go b/test/e2e/mnist_rayjob_raycluster_test.go index b7f039d9..5bf5e932 100644 --- a/test/e2e/mnist_rayjob_raycluster_test.go +++ b/test/e2e/mnist_rayjob_raycluster_test.go @@ -313,7 +313,7 @@ func constructRayJob(_ Test, namespace *corev1.Namespace, rayCluster *rayv1.RayC Entrypoint: "python /home/ray/jobs/mnist.py", RuntimeEnvYAML: ` pip: - - pytorch_lightning==1.5.10 + - pytorch_lightning==1.9.5 - torchmetrics==0.9.1 - torchvision==0.12.0 env_vars: diff --git a/test/odh/resources/requirements.txt b/test/odh/resources/requirements.txt index cf9fb588..a58df295 100644 --- a/test/odh/resources/requirements.txt +++ b/test/odh/resources/requirements.txt @@ -1,6 +1,6 @@ {{.PipIndexUrl}} {{.PipTrustedHost}} -pytorch_lightning==1.5.10 +pytorch_lightning==1.9.5 ray_lightning torchmetrics==0.9.1 torchvision==0.12.0