diff --git a/Dockerfile b/Dockerfile new file mode 100644 index 0000000..49c1b24 --- /dev/null +++ b/Dockerfile @@ -0,0 +1,19 @@ +FROM docker.io/nvidia/cuda:12.0.0-cudnn8-runtime-ubuntu22.04 + +RUN apt-get update && \ + apt-get install -y ffmpeg python3 python3-pip && \ + apt-get clean && \ + rm -rf /var/lib/apt/lists/* + +WORKDIR /app + +COPY requirements.txt . + +RUN pip3 install --no-cache-dir -r requirements.txt + +COPY . . + +EXPOSE 5000 + +# 启动 whisper_fastapi.py +ENTRYPOINT ["python3", "whisper_fastapi.py"] diff --git a/start-docker.sh b/start-docker.sh new file mode 100755 index 0000000..dee552a --- /dev/null +++ b/start-docker.sh @@ -0,0 +1,10 @@ +#!/bin/bash + +docker run -d --name whisper-fastapi \ + --restart unless-stopped \ + --name whisper-fastapi \ + -v ~/.cache/huggingface:/root/.cache/huggingface \ + --gpus all \ + -p 5000:5000 \ + docker.io/heimoshuiyu/whisper-fastapi:lastet \ + --model large-v2 \ No newline at end of file