23 lines
467 B
Bash
Executable File
23 lines
467 B
Bash
Executable File
#!/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
|