Update colossalai version in Dockerfile (#16766)

update docker

Author: awaelchli

dockers/base-cuda/Dockerfile

@@ -152,7 +152,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