feat: torch container startup script

2026-06-11 11:51:30 +02:00
parent aad37ba6eb
commit 59f6f34566
2 changed files with 20 additions and 0 deletions
@@ -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