fix: Allow custom user ID

[yurkovak]

A fix for #1130. It's not possible to write into a volume with a wrong user making it more hassle to train.

[tancik]

LGTM

[Ilyabasharov]

Duplicate of prev PR. Current pr works not all cases.

[XinyueZ]

The PR has problem.
I was trying to build the container after merged (current main), I got error:

 > [ 6/22] RUN useradd -m -d /home/user -g root -G sudo -u ${USER_ID} user:                                                                                                                                            
#0 0.447 useradd: invalid user ID 'user'                                                                                                                                                                               
------                                                                                                                                                                                                                 
Dockerfile:103                                                                                                                                                                                                         
--------------------
 101 |     
 102 |     # Create non root user and setup environment.
 103 | >>> RUN useradd -m -d /home/user -g root -G sudo -u ${USER_ID} user
 104 |     RUN usermod -aG sudo user
 105 |     # Set user password