Update colossalai version in Dockerfile (#16766)

update docker
Lightning-AI · Feb 15, 2023 · 3b7f186 · 3b7f186
1 parent 0628ec8
commit 3b7f186
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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