From bfc4694db45ff1517aeab8ad952602c4086abb1d Mon Sep 17 00:00:00 2001 From: Enrico Minack Date: Mon, 21 Mar 2022 23:30:46 +0100 Subject: [PATCH] Also move head images to 1.5.9 Signed-off-by: Enrico Minack --- docker-compose.test.yml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docker-compose.test.yml b/docker-compose.test.yml index db2afce6f8..053cdf91db 100644 --- a/docker-compose.test.yml +++ b/docker-compose.test.yml @@ -94,7 +94,7 @@ services: KERAS_PACKAGE: None PYTORCH_PACKAGE: torch-nightly TORCHVISION_PACKAGE: torchvision - PYTORCH_LIGHTNING_PACKAGE: pytorch-lightning + PYTORCH_LIGHTNING_PACKAGE: pytorch-lightning==1.5.9 MXNET_PACKAGE: mxnet-nightly # these are the lowest framework versions that Horovod compiles with, but they are not tested test-cpu-gloo-py3_7-tfmin-kerasmin-torchmin-mxnetmin-pysparkmin: @@ -224,7 +224,7 @@ services: TENSORFLOW_PACKAGE: tf-nightly-gpu KERAS_PACKAGE: None PYTORCH_PACKAGE: torch-nightly-cu111 - PYTORCH_LIGHTNING_PACKAGE: pytorch-lightning + PYTORCH_LIGHTNING_PACKAGE: pytorch-lightning==1.5.9 TORCHVISION_PACKAGE: torchvision MXNET_PACKAGE: mxnet-nightly-cu112 # these are the lowest framework versions that Horovod compiles with, but they are not tested