From 677bf597c99df952233e81032482028350617e5d Mon Sep 17 00:00:00 2001 From: liuzeming Date: Mon, 27 Mar 2023 21:43:18 +0800 Subject: [PATCH 1/2] [docker] Add ARG VERSION to determine the Tag --- docker/Dockerfile | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/docker/Dockerfile b/docker/Dockerfile index 49ff9b344268..52c7bf5601c6 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -15,7 +15,8 @@ RUN git clone https://github.com/NVIDIA/apex && \ pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" --global-option="--fast_layer_norm" ./ # install colossalai -RUN git clone https://github.com/hpcaitech/ColossalAI.git \ +ARG VERSION=1 +RUN git clone -b ${VERSION} https://github.com/hpcaitech/ColossalAI.git \ && cd ./ColossalAI \ && CUDA_EXT=1 pip install -v --no-cache-dir . From 9f7754357d9570580fda6a4a25871aa2d6f0fc8f Mon Sep 17 00:00:00 2001 From: liuzeming Date: Mon, 27 Mar 2023 21:43:38 +0800 Subject: [PATCH 2/2] [workflow] fixed the version in the release docker workflow --- .github/workflows/release_docker_after_merge.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/release_docker_after_merge.yml b/.github/workflows/release_docker_after_merge.yml index 607c19b05472..6b6684641117 100644 --- a/.github/workflows/release_docker_after_merge.yml +++ b/.github/workflows/release_docker_after_merge.yml @@ -26,7 +26,7 @@ jobs: run: | version=$(cat version.txt) tag=hpcaitech/colossalai:$version - docker build --build-arg http_proxy=http://172.17.0.1:7890 --build-arg https_proxy=http://172.17.0.1:7890 -t $tag ./docker + docker build --build-arg http_proxy=http://172.17.0.1:7890 --build-arg https_proxy=http://172.17.0.1:7890 --build-arg VERSION=v${version} -t $tag ./docker echo "tag=${tag}" >> $GITHUB_OUTPUT - name: Log in to Docker Hub