add Docker with cuda 12.1 as Pytorch update libs

Author: quy-ng

Dockerfile

@@ -0,0 +1,32 @@
+FROM mambaorg/micromamba:jammy-cuda-12.1.0
+
+ENV LANG=C.UTF-8 LC_ALL=C.UTF-8
+ENV ENV_NAME=diffusors SHELL=/bin/bash
+ARG NEW_MAMBA_USER=$ENV_NAME
+ARG NEW_MAMBA_USER_ID=5005
+ARG NEW_MAMBA_USER_GID=5005
+EXPOSE 8888
+
+RUN micromamba install -y -n base -c conda-forge python=3.11 \
+    && micromamba clean --all --yes
+
+USER root
+RUN apt update && apt install -y git tmux nvtop htop \
+    && apt clean autoremove -y
+RUN usermod "--login=${NEW_MAMBA_USER}" "--home=/home/${NEW_MAMBA_USER}" \
+        --move-home "-u ${NEW_MAMBA_USER_ID}" "${MAMBA_USER}" && \
+    groupmod "--new-name=${NEW_MAMBA_USER}" \
+        "-g ${NEW_MAMBA_USER_GID}" "${MAMBA_USER}" && \
+    # Update the expected value of MAMBA_USER for the
+    # _entrypoint.sh consistency check.
+    echo "${NEW_MAMBA_USER}" > "/etc/arg_mamba_user" && :
+ENV MAMBA_USER=$NEW_MAMBA_USER
+
+USER $MAMBA_USER
+WORKDIR /workspace
+# Install any needed packages specified in environment.yml
+COPY environment.yml environment.yml
+RUN micromamba create -n $ENV_NAME -f environment.yml -y \
+    && micromamba clean --all --yes
+
+CMD ["jupyter", "lab", "--ip=0.0.0.0", "--no-browser"]

run_docker.sh

@@ -0,0 +1,25 @@
+#!/bin/bash
+
+branch=$(git rev-parse --abbrev-ref HEAD)
+
+docker rm -f diffusors-${branch}
+docker build --build-arg NEW_MAMBA_USER_ID=$(id -u) --build-arg NEW_MAMBA_USER_GID=$(id -g)  -t diffusors:${branch} .
+docker run --gpus all \
+    --env-file .env \
+    -it -d -v $PWD:/workspace \
+    -v /data:/data \
+    -p 8888:8888 \
+    --shm-size=200g --ulimit memlock=-1 --ulimit stack=64000000 \
+    --name diffusors-${branch} \
+    diffusors:${branch} bash
+
+# Wait for the Docker container to start
+sleep 5
+
+# Get the logs from the running Docker container
+logs=$(docker logs diffusors-${branch})
+
+# Extract the token from the logs
+token=$(echo "${logs}" | grep -oP '(?<=token=)[a-z0-9]*' | head -n 1)
+
+echo "Jupyter Notebook Token: ${token}"