datasci/verbanote-server
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Use cuda-enabled base image

Browse source 
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
Download patch file Download diff file Expand all files Collapse all files

5
Dockerfile
View file

@ -1,4 +1,4 @@
ARG PYTHON_VERSION=3.11.0
ARG PYTHON_VERSION=3.10.0
ARG POETRY_VERSION=1.5.0
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_PATH=/verbanote
ARG VERBANOTE_OUTPUT_PATH=/out