#!/bin/sh

name=$([[ $1 ]] && echo $1 || echo "pytorch")
mount=$([[ $2 ]] && echo $2 || echo "$HOME/projects/")

echo "
=> Mounting folder to /mnt
=> Starting container
"

docker run -it \
    --device=/dev/kfd \
    --device=/dev/dri \
    --ipc=host \
    --shm-size 16G \
    --group-add video \
    --cap-add=SYS_PTRACE \
    --security-opt seccomp=unconfined \
    -v $mount:/mnt \
	--hostname jh-dev-container-$name \
    --name $name \
    rocm/pytorch:latest
