#!/bin/bash
set -e

if [[ "$EUID" == "0" ]]; then
    echo "Please run as your own user (i.e., DO NOT sudo)."
    exit 1
fi
container_name=$USER_$(date +%y%m%d-%H%M%S)
image_name=${1:-"nasp-ubuntu"}
cmd=${2:-"/etc/startup.sh"}
echo "container name: $container_name"
echo "image: $image_name"
sudo docker run -d --net=host -v /home2:/home2 \
    --cap-add=SYS_NICE --cap-add=IPC_LOCK \
    --device=/dev/infiniband/rdma_cm --device=/dev/infiniband/issm0 \
    --device=/dev/infiniband/umad0 --device=/dev/infiniband/uverbs0 \
    --runtime=nvidia -e NVIDIA_VISIBLE_DEVICES=all \
    -v /usr/local/cuda:/usr/local/cuda --security-opt \
    seccomp=unconfined --ulimit memlock=-1:-1 \
    --restart=always \
    --name $container_name -it $image_name $cmd