From 79f082d6dfb6dc2d05fc40761728fc89b04be15e Mon Sep 17 00:00:00 2001 From: vfdev Date: Thu, 7 Nov 2024 14:53:52 +0100 Subject: [PATCH] Fixed torch version in docker push_all.sh (#3300) * Updated pytorch version in CI and docker * Fix docker tag * Fix docker tag value removing +cu124 like words from torch version * Fix version compare in test_image * Fixed torch version in docker push_all.sh --- docker/push_all.sh | 12 ++---------- 1 file changed, 2 insertions(+), 10 deletions(-) diff --git a/docker/push_all.sh b/docker/push_all.sh index ae8a84d4472..6951ff80c9b 100644 --- a/docker/push_all.sh +++ b/docker/push_all.sh @@ -26,11 +26,9 @@ echo $DOCKER_TOKEN | docker login --username=$DOCKER_USER --password-stdin set -xeu - if [ ${push_selected_image} == "all" ]; then - image_name="base" - image_tag=`docker run --rm -i pytorchignite/${image_name}:latest python -c "import torch; import ignite; print(torch.__version__ + \"-\" + ignite.__version__, end=\"\")"` + image_tag=`docker run --rm -i pytorchignite/${image_name}:latest python -c "import torch; import ignite; print(torch.__version__.split('+')[0] + \"-\" + ignite.__version__, end=\"\")"` for image_name in "base" "vision" "nlp" "apex" "apex-vision" "apex-nlp" do @@ -40,9 +38,6 @@ if [ ${push_selected_image} == "all" ]; then done - image_name="hvd-base" - image_tag=`docker run --rm -i pytorchignite/${image_name}:latest python -c "import torch; import ignite; print(torch.__version__ + \"-\" + ignite.__version__, end=\"\")"` - for image_name in "hvd-base" "hvd-vision" "hvd-nlp" "hvd-apex" "hvd-apex-vision" "hvd-apex-nlp" do @@ -52,9 +47,6 @@ if [ ${push_selected_image} == "all" ]; then done # DEPRECATED due to no activity - # image_name="msdp-apex" - # image_tag=`docker run --rm -i pytorchignite/${image_name}:latest python -c "import torch; import ignite; print(torch.__version__ + \"-\" + ignite.__version__, end=\"\")"` - # for image_name in "msdp-apex" "msdp-apex-vision" "msdp-apex-nlp" # do @@ -66,7 +58,7 @@ if [ ${push_selected_image} == "all" ]; then else image_name=${push_selected_image} - image_tag=`docker run --rm -i pytorchignite/${image_name}:latest python -c "import torch; import ignite; print(torch.__version__ + \"-\" + ignite.__version__, end=\"\")"` + image_tag=`docker run --rm -i pytorchignite/${image_name}:latest python -c "import torch; import ignite; print(torch.__version__.split('+')[0] + \"-\" + ignite.__version__, end=\"\")"` docker push pytorchignite/${image_name}:latest docker push pytorchignite/${image_name}:${image_tag}