Update colossalai version in Dockerfile (#16766)

update docker
2023-02-15 20:20:13 +01:00 · 2023-02-15 20:20:13 +01:00 · 3b7f186a05
parent 0628ec83c4
commit 3b7f186a05
1 changed files with 1 additions and 1 deletions
--- a/dockers/base-cuda/Dockerfile
+++ b/dockers/base-cuda/Dockerfile
@ -102,7 +102,7 @@ RUN \
    # install ColossalAI
    # TODO: 1.13 wheels are not released, remove skip once they are
    if [[ $PYTORCH_VERSION != "1.13" ]]; then \
-        pip install "colossalai==0.2.3"; \
+        pip install "colossalai==0.2.4"; \
        python -c "import colossalai; print(colossalai.__version__)" ; \
    fi