docker.sh

#!/bin/bash
#set -x
dry_run=0
interactive=0
grep_value=
command=
it=" -it"
USER=${USER:-$(whoami)}
image=${USER}_vllm
name=${USER}_vllm

while [[ $# -gt 0 ]] ; do
  i=$1
  case $i in
  -n|--name)
    name="$2"
    shift
  ;;
  -d|--dry=run)
    dry_run=1
  ;;
  -g|--grep)
    grep_value="$2"
    shift
  ;;
  -i|--interactive)
    interactive=1
  ;;
  --noit)
    it=
  ;;
  -c|--command)
    command="$2"
    shift
  ;;
  *)
    image=$1
  ;;
  esac
  shift
done

if [[ $interactive == 1 ]] ; then
    grep_arg=
    if [[ $grep_value != "" ]] ; then
      grep_arg=" | grep $grep_value"
    fi
    cmd="docker images | awk '{print \$1 \":\" \$2}' | grep -v none | grep -v REPOSITORY:TAG $grep_arg"
    images=$(eval $cmd)
    i=0
    for im in $images ; do
    echo "$i $im"
    i=$((i+1))
    done
    read -p "Select image: " selection
    images_arr=($images)
    image=${images_arr[$selection]}
    read -p "Container name: " name
fi
echo "Image: $image"

name_arg=
if [[ $name != "" ]] ; then
    name_arg=" --name $name"
fi

if [[ $dry_run != 1 ]] ; then
tmux rename-window "Docker:$name"
docker run ${it} --rm --device=/dev/kfd --device=/dev/dri --mount type=bind,source=/home/gshtrasb/Projects,target=/projects --mount type=bind,source=/data/models,target=/models --ulimit core=0:0 --group-add video --cap-add=SYS_PTRACE $name_arg $image $command
tmux setw automatic-rename on
echo "Finished docker image $image"
else
echo "docker run ${it} --rm --device=/dev/kfd --device=/dev/dri --mount type=bind,source=/home/gshtrasb/Projects,target=/projects --mount type=bind,source=/data/models,target=/models --ulimit core=0:0 --group-add video --cap-add=SYS_PTRACE $name_arg $image" $command
fi