2022-02-20 06:42:10 +02:00
|
|
|
## GPU Build
|
2022-02-20 06:57:11 +02:00
|
|
|
# FROM tensorflow/tensorflow:latest-gpu as gpu
|
2022-02-20 06:42:10 +02:00
|
|
|
|
2022-02-20 06:57:11 +02:00
|
|
|
# WORKDIR /code
|
2022-02-20 06:42:10 +02:00
|
|
|
|
2022-02-20 06:57:11 +02:00
|
|
|
# COPY ./requirements.txt /code/requirements.txt
|
2022-02-20 06:42:10 +02:00
|
|
|
|
2022-02-20 06:57:11 +02:00
|
|
|
# RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
|
2022-02-20 06:42:10 +02:00
|
|
|
|
2022-02-20 06:57:11 +02:00
|
|
|
# COPY ./app /code/app
|
2022-02-20 06:42:10 +02:00
|
|
|
|
|
|
|
|
|
|
|
## CPU BUILD
|
|
|
|
FROM python:3.8 as cpu
|
|
|
|
|
2022-03-19 03:23:05 +02:00
|
|
|
RUN apt-get update
|
|
|
|
RUN apt-get install ffmpeg libsm6 libxext6 -y
|
|
|
|
|
2022-02-20 06:42:10 +02:00
|
|
|
WORKDIR /code
|
|
|
|
|
|
|
|
COPY ./requirements.txt /code/requirements.txt
|
|
|
|
|
|
|
|
RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
|
|
|
|
|
|
|
|
COPY ./app /code/app
|