Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /tmp; ExposedPorts 6006/tcp, 8888/tcp

GID=1000 UID=1000 GROUP=ubuntu USER=ubuntu if [ "$UID" -eq 1000 ]; then if [ "$USER" != "ubuntu" ]; then usermod --login $USER --home /home/$USER ubuntu &&
  groupmod --new-name $USER ubuntu &&
  usermod --uid $UID --gid $GID $USER &&
  mv /home/ubuntu /home/$USER &&
  chown -R $UID:$GID /home/$USER; else usermod --uid $UID --gid $GID $USER &&
  chown -R $UID:$GID /home/$USER; fi; else groupadd -f -g ${GID} ${GROUP} &&
  useradd -G sudo -g ${GID} -u ${UID} -m ${USER} &&
  echo ${USER}:${USER} | chpasswd; fi

GID=1000 UID=1000 GROUP=ubuntu USER=ubuntu echo root:root | chpasswd

USER=ubuntu

GROUP=ubuntu

UID=1000

GID=1000

rm -rf /tmp/drive-llm

pip install transformers datasets

Llama-3.1-8B-INT4-ONNX/ /tmp/Llama-3.1-8B-INT4-ONNX/