From 59f6f345667aa7e894bdb129fca7d9ff5c4e3f37 Mon Sep 17 00:00:00 2001 From: Janis Hutz Date: Thu, 11 Jun 2026 11:51:30 +0200 Subject: [PATCH] feat: torch container startup script --- torch-rocm/add-container.sh | 20 ++++++++++++++++++++ torch-rocm/docker-compose.yml | 0 2 files changed, 20 insertions(+) create mode 100755 torch-rocm/add-container.sh delete mode 100644 torch-rocm/docker-compose.yml diff --git a/torch-rocm/add-container.sh b/torch-rocm/add-container.sh new file mode 100755 index 0000000..cd77a9f --- /dev/null +++ b/torch-rocm/add-container.sh @@ -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 diff --git a/torch-rocm/docker-compose.yml b/torch-rocm/docker-compose.yml deleted file mode 100644 index e69de29..0000000