Slide 14
Slide 14 text
Two different Tensorflows???
FROM nvidia/cuda:9.0-cudnn7-runtime-ubuntu16.04 AS base
COPY ./Pipfile ./Pipfile.lock ./
RUN pipenv install --deploy --dev
CMD ["pipenv", "run", "python", "-m", "my.worker"]
FROM base AS base-gpu
RUN pipenv run pip install tensorflow-gpu==$(pipenv run pip show tensorflow | grep '^Version' | awk '{print $2}')
FROM base as app
COPY . .
FROM base-gpu as gpu
COPY . .