80 lines
1.7 KiB
Bash
Executable File
80 lines
1.7 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
set -e
|
|
|
|
trap 'echo "An error occurred. Please contact the manager." >&2' ERR
|
|
|
|
echo "================== dnew ==================="
|
|
echo "=== contact: xd21@mails.tsinghua.edu.cn ==="
|
|
|
|
test_not_empty()
|
|
{
|
|
if [[ -z "$1" ]]; then
|
|
echo $2 >&2
|
|
exit
|
|
fi
|
|
}
|
|
|
|
ask_for_yN()
|
|
{
|
|
if [[ "$DFS_QUIET" == "1" ]]; then
|
|
echo 0
|
|
else
|
|
read -p "$1 [yN]: " yn
|
|
case $yn in
|
|
[Yy]* ) echo 1;;
|
|
* ) echo 0;;
|
|
esac
|
|
fi
|
|
}
|
|
|
|
container_name=$(whoami)_$(date +%y%m%d-%H%M%S)
|
|
|
|
base_command="sudo docker run -d --net=host \\
|
|
-v /home2:/home2 -v /gshare:/gshare \\
|
|
--privileged \\
|
|
--security-opt seccomp=unconfined --ulimit memlock=-1:-1 \\
|
|
--restart=unless-stopped \\
|
|
--name $container_name"
|
|
|
|
|
|
if [[ "$EUID" == "0" ]]; then
|
|
echo "Please run as your own user (i.e., DO NOT sudo)."
|
|
exit 1
|
|
fi
|
|
|
|
ret=$(ask_for_yN "Use GPU?")
|
|
if [[ "$ret" == "1" ]]; then
|
|
base_command="$base_command \\
|
|
--device nvidia.com/gpu=all"
|
|
fi
|
|
ret=$(ask_for_yN "Use RDMA?")
|
|
if [[ "$ret" == "1" ]]; then
|
|
base_command="$base_command \\
|
|
--device=/dev/infiniband/uverbs0"
|
|
fi
|
|
|
|
image_name="git.nasp.fit/nasp/nasp-ubuntu"
|
|
base_command="$base_command \\
|
|
-it $image_name"
|
|
|
|
cmd="/etc/startup.sh"
|
|
base_command="$base_command $cmd"
|
|
|
|
echo "Will run:"
|
|
echo "====="
|
|
echo "$base_command"
|
|
echo "====="
|
|
ret=$(ask_for_yN "Start the container?")
|
|
|
|
if [[ "$ret" == "1" ]]; then
|
|
eval "$base_command"
|
|
echo "Container $container_name started. You can use the following command to get in:"
|
|
echo "sudo docker exec -it $container_name bash"
|
|
if [[ "$image_name" == "git.nasp.fit/nasp/nasp-ubuntu" ]]; then
|
|
sleep 1s
|
|
sudo docker exec -it $container_name code.sh
|
|
fi
|
|
else
|
|
echo "Aborted."
|
|
fi
|