Spaces:
Sleeping
Sleeping
j
commited on
Commit
·
c770543
1
Parent(s):
1d8b8f7
added GPU dockerfile
Browse files- Dockerfile +14 -2
Dockerfile
CHANGED
@@ -1,25 +1,32 @@
|
|
1 |
FROM swaggerapi/swagger-ui:v4.18.2 AS swagger-ui
|
2 |
-
FROM
|
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
|