chore(docker): improve logging setup

- Implemented rich logging in Dockerfile and set FORCE_COLOR environment variable to true
- Set the COLUMNS environment variable to `100` in Dockerfile for wider output
- Replaced single quotes with double quotes for `device_ids` value in docker-compose.yaml

Signed-off-by: 陳鈞 <jim60105@gmail.com>

Dockerfile

@@ -112,6 +112,10 @@ ENV PYTHONPATH="${PYTHONPATH}:/home/$UID/.local/lib/python3.10/site-packages"
 ENV LD_LIBRARY_PATH="/usr/local/cuda/lib:/usr/local/cuda/lib64:${LD_LIBRARY_PATH}"
 ENV LD_PRELOAD=libtcmalloc.so
 ENV PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=python
+# Rich logging
+# https://rich.readthedocs.io/en/stable/console.html#interactive-mode
+ENV FORCE_COLOR="true"
+ENV COLUMNS="100"
 
 WORKDIR /app
 

docker-compose.yaml

@@ -36,8 +36,8 @@ services:
           devices:
             - driver: nvidia
               capabilities: [gpu]
-              device_ids: ['all']
-  
+              device_ids: ["all"]
+
   tensorboard:
     container_name: tensorboard
     image: tensorflow/tensorflow:latest-gpu
@@ -52,4 +52,4 @@ services:
           devices:
             - driver: nvidia
               capabilities: [gpu]
-              device_ids: ['all']
+              device_ids: ["all"]