20 lines
702 B
Plaintext
20 lines
702 B
Plaintext
|
#!/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"}
|
||
|
|
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 \
|
||
|
|
--name $container_name -it $image_name bash
|