j commited on
Commit
c770543
·
1 Parent(s): 1d8b8f7

added GPU dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +14 -2
Dockerfile CHANGED
@@ -1,25 +1,32 @@
1
  FROM swaggerapi/swagger-ui:v4.18.2 AS swagger-ui
2
- FROM python:3.10-slim
3
 
4
  ARG SERVICE_USER=service
5
  ARG SERVICE_UID=1000
6
  ARG SERVICE_GID=1000
7
 
 
8
  ENV POETRY_VENV=/app/.venv
9
 
10
  RUN export DEBIAN_FRONTEND=noninteractive \
11
  && apt-get -qq update \
12
  && apt-get -qq install --no-install-recommends \
 
 
 
13
  lua5.3 \
14
  lua5.4 \
15
  lua-check \
16
  fswatch \
17
  make \
18
- cargo \
19
  ffmpeg \
20
  redis \
21
  && rm -rf /var/lib/apt/lists/*
22
 
 
 
 
 
23
  RUN groupadd -g $SERVICE_GID $SERVICE_USER || true
24
  RUN useradd -u $SERVICE_UID -g $SERVICE_GID -d /app -s /usr/sbin/nologin $SERVICE_USER || true
25
 
@@ -38,8 +45,13 @@ RUN python3 -m venv $POETRY_VENV \
38
 
39
  ENV PATH="${PATH}:${POETRY_VENV}/bin"
40
 
 
 
41
  RUN poetry config virtualenvs.in-project true
 
 
42
  RUN poetry install && rm -rf /app/.cache/pypoetry
 
43
 
44
  WORKDIR /app/reascripts/ReaSpeech
45
  RUN make publish
 
1
  FROM swaggerapi/swagger-ui:v4.18.2 AS swagger-ui
2
+ FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04
3
 
4
  ARG SERVICE_USER=service
5
  ARG SERVICE_UID=1000
6
  ARG SERVICE_GID=1000
7
 
8
+ ENV PYTHON_VERSION=3.10
9
  ENV POETRY_VENV=/app/.venv
10
 
11
  RUN export DEBIAN_FRONTEND=noninteractive \
12
  && apt-get -qq update \
13
  && apt-get -qq install --no-install-recommends \
14
+ python${PYTHON_VERSION} \
15
+ python${PYTHON_VERSION}-venv \
16
+ python3-pip \
17
  lua5.3 \
18
  lua5.4 \
19
  lua-check \
20
  fswatch \
21
  make \
 
22
  ffmpeg \
23
  redis \
24
  && rm -rf /var/lib/apt/lists/*
25
 
26
+ RUN ln -s -f /usr/bin/python${PYTHON_VERSION} /usr/bin/python3 && \
27
+ ln -s -f /usr/bin/python${PYTHON_VERSION} /usr/bin/python && \
28
+ ln -s -f /usr/bin/pip3 /usr/bin/pip
29
+
30
  RUN groupadd -g $SERVICE_GID $SERVICE_USER || true
31
  RUN useradd -u $SERVICE_UID -g $SERVICE_GID -d /app -s /usr/sbin/nologin $SERVICE_USER || true
32
 
 
45
 
46
  ENV PATH="${PATH}:${POETRY_VENV}/bin"
47
 
48
+ COPY --chown=$SERVICE_UID:$SERVICE_GID poetry.lock pyproject.toml ./
49
+
50
  RUN poetry config virtualenvs.in-project true
51
+ RUN poetry install --no-root
52
+
53
  RUN poetry install && rm -rf /app/.cache/pypoetry
54
+ RUN $POETRY_VENV/bin/pip install --no-cache-dir torch==1.13.1+cu117 -f https://download.pytorch.org/whl/torch
55
 
56
  WORKDIR /app/reascripts/ReaSpeech
57
  RUN make publish