[log] better logging display with rich (#426)

* better logger using rich

* remove deepspeed in zero requirements

colossalai/logging/logging.py

@@ -5,11 +5,12 @@ import colossalai
 import logging
 from pathlib import Path
 from typing import Union
+from rich.logging import RichHandler
 
 from colossalai.context.parallel_mode import ParallelMode
 
 _FORMAT = 'colossalai - %(name)s - %(asctime)s %(levelname)s: %(message)s'
-logging.basicConfig(level=logging.INFO, format=_FORMAT)
+logging.basicConfig(level=logging.INFO, format=_FORMAT, handlers=[RichHandler()])
 
 
 class DistributedLogger:

requirements/requirements-test.txt

@@ -1,3 +1,3 @@
 pytest
 rpyc
-matplotlib
+matplotlib

requirements/requirements-zero.txt

@@ -1 +0,0 @@
-deepspeed

requirements/requirements.txt

@@ -6,3 +6,4 @@ psutil
 tensorboard
 packaging
 pre-commit
+rich