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