diff --git a/Dockerfile b/Dockerfile index 39db138..7fb96b5 100644 --- a/Dockerfile +++ b/Dockerfile @@ -3,7 +3,7 @@ ARG GPU=nvidia FROM nvidia/cuda:13.0.2-cudnn-runtime-ubuntu24.04 AS base-nvidia -FROM rocm/dev-ubuntu-24.04:7.2.1-complete AS base-amd +FROM rocm/dev-ubuntu-24.04:7.2.1 AS base-amd FROM base-${GPU} AS app diff --git a/README.md b/README.md index 476fb8c..afd0d39 100644 --- a/README.md +++ b/README.md @@ -31,14 +31,14 @@ cp input.mkv "$WORK_DIR/" # nvidia docker run --rm --gpus all -e AI_API_KEY \ - --mount type=volume,src=sumka-whisper-cache,dst=/root/.cache/whisper \ + --mount type=volume,src=whisper-cache,dst=/tmp/sumka-cache/whisper \ --mount type=bind,src="$WORK_DIR",dst=/work \ --tmpfs /tmp:rw,size=8g sumka:nvidia # amd docker run --rm --device=/dev/kfd --device=/dev/dri --group-add video \ --security-opt seccomp=unconfined -e AI_API_KEY \ - --mount type=volume,src=sumka-whisper-cache,dst=/root/.cache/whisper \ + --mount type=volume,src=whisper-cache,dst=/tmp/sumka-cache/whisper \ --mount type=bind,src="$WORK_DIR",dst=/work \ --tmpfs /tmp:rw,size=8g sumka:amd ``` diff --git a/start_amd.sh b/start_amd.sh new file mode 100755 index 0000000..5449d9d --- /dev/null +++ b/start_amd.sh @@ -0,0 +1,5 @@ +#!/usr/bin/bash + +mkdir runtime &> /dev/null + +docker run --device=/dev/kfd --device=/dev/dri --group-add video --security-opt seccomp=unconfined -e AI_API_KEY --mount type=bind,src=./runtime,dst=/work --mount type=volume,src=whisper-cache,dst=/tmp/sumka-cache/whisper --tmpfs /tmp:rw,size=8g sumka:amd --resolver-ai-model google/gemini-3.8-flash --structure-ai-model google/gemini-3.8-flash --refiner-ai-model google/gemini-3.8-flash