feat: torch container startup script
This commit is contained in:
Executable
+20
@@ -0,0 +1,20 @@
|
||||
#!/bin/sh
|
||||
|
||||
containername=$([[ $1 ]] && echo $1 || echo "pytorch")
|
||||
containermount=$([[ $2 ]] && echo $2 || echo "$HOME/projects/")
|
||||
|
||||
echo "
|
||||
=> Mounting folder to /mnt
|
||||
"
|
||||
|
||||
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 $containermount:/mnt \
|
||||
--name $containername \
|
||||
rocm/pytorch:latest
|
||||
Reference in New Issue
Block a user