Added docker support
This commit is contained in:
35
README.md
35
README.md
@@ -15,7 +15,36 @@
|
||||
|
||||
## Установка (Docker)
|
||||
|
||||
TODO
|
||||
Образы собираются из одного Dockerfile:
|
||||
|
||||
```bash
|
||||
docker build --build-arg GPU=nvidia -t sumka:nvidia .
|
||||
docker build --build-arg GPU=amd -t sumka:amd .
|
||||
```
|
||||
|
||||
Для NVIDIA нужен настроенный NVIDIA Container Toolkit. Для AMD на хосте должен
|
||||
быть установлен совместимый ROCm. Рабочая директория монтируется в `/work`:
|
||||
|
||||
```bash
|
||||
WORK_DIR=$(mktemp -d)
|
||||
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=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=bind,src="$WORK_DIR",dst=/work \
|
||||
--tmpfs /tmp:rw,size=8g sumka:amd
|
||||
```
|
||||
|
||||
Все входные, промежуточные и итоговые файлы находятся в `WORK_DIR`. Для дампов
|
||||
запросов заранее создайте `"$WORK_DIR/debug"`; промпты остаются внутри образа.
|
||||
|
||||
## Установка (Python)
|
||||
|
||||
@@ -28,8 +57,8 @@ cd 2026-linux-sumka
|
||||
python3 -m venv .venv
|
||||
# Активировать виртуальное окружение
|
||||
. .venv/bin/activate
|
||||
# Установить зависимости
|
||||
pip install -r requirements.txt
|
||||
# Установить зависимости под свою видеокарту
|
||||
pip install -r requirements-nvidia.txt # либо requirements-amd.txt
|
||||
|
||||
# Также должны быть установлены ffmpeg и Chromium/Google Chrome
|
||||
|
||||
|
||||
Reference in New Issue
Block a user