Use cuda-enabled base image

Since I can't build cuda images locally (without nvidia gpu), we make use
of an existing one. That also means downgrading python to Debian's version
3.10.
This commit is contained in:
Marty Oehme 2023-08-22 20:35:25 +02:00
parent 56fa2eca25
commit 3246469be2
Signed by: Marty
GPG key ID: EDBF2ED917B2EF6A
2 changed files with 4 additions and 6 deletions

View file

@ -1,4 +1,4 @@
ARG PYTHON_VERSION=3.11.0 ARG PYTHON_VERSION=3.10.0
ARG POETRY_VERSION=1.5.0 ARG POETRY_VERSION=1.5.0
FROM python:${PYTHON_VERSION} as staging FROM python:${PYTHON_VERSION} as staging
@ -31,7 +31,8 @@ RUN poetry install
# -------------- # --------------
FROM staging as worker FROM ghcr.io/pytorch/pytorch-nightly:668af07-cu11.8.0 as worker
ARG APP_NAME=verbanote ARG APP_NAME=verbanote
ARG APP_PATH=/verbanote ARG APP_PATH=/verbanote
ARG VERBANOTE_OUTPUT_PATH=/out ARG VERBANOTE_OUTPUT_PATH=/out

View file

@ -7,18 +7,15 @@ readme = "README.md"
# packages = [{ include = "verbanote" }] # packages = [{ include = "verbanote" }]
[tool.poetry.dependencies] [tool.poetry.dependencies]
python = "^3.11" python = "^3.10"
requests = "^2.31.0" requests = "^2.31.0"
static-ffmpeg = "^2.5" static-ffmpeg = "^2.5"
runpod = "^1.1.3" runpod = "^1.1.3"
pydub = "^0.25.1" pydub = "^0.25.1"
#light-the-torch = "^0.7.5"
light-the-torch = "^0.7.5" light-the-torch = "^0.7.5"
hmmlearn = {git = "https://github.com/hmmlearn/hmmlearn.git"} hmmlearn = {git = "https://github.com/hmmlearn/hmmlearn.git"}
pyannote-audio = {git = "https://github.com/pyannote/pyannote-audio.git", rev = "develop"} pyannote-audio = {git = "https://github.com/pyannote/pyannote-audio.git", rev = "develop"}
#openai-whisper = { git = "https://github.com/openai/whisper.git" } #openai-whisper = { git = "https://github.com/openai/whisper.git" }
#hmmlearn = {git = "https://github.com/hmmlearn/hmmlearn.git"}
#pyannote-audio = {git = "https://github.com/pyannote/pyannote-audio.git", rev = "develop"}
[build-system] [build-system]
requires = ["poetry-core"] requires = ["poetry-core"]