Commit ae247272 authored by Vish Kannan's avatar Vish Kannan Committed by Karmel Allison
Browse files

Adding dockerfiles to generate official docker images that can used for...


Adding dockerfiles to generate official docker images that can used for running models in this repo on container based platforms (#3399)
Signed-off-by: default avatarVishnu kannan <vishnuk@google.com>
parent a41f00ac
# Docker image for running examples in Tensorflow models.
# base_image depends on whether we are running on GPUs or non-GPUs
FROM ubuntu:latest
RUN apt-get update && apt-get install -y --no-install-recommends \
ca-certificates \
build-essential \
git \
python \
python-pip \
python-setuptools
RUN pip install tf-nightly
# Checkout tensorflow/models and pin to a particular version
RUN git clone https://github.com/tensorflow/models.git /tensorflow_models
# Docker image for running examples in Tensorflow models.
# base_image depends on whether we are running on GPUs or non-GPUs
FROM nvidia/cuda:9.0-cudnn7-runtime-ubuntu16.04
RUN apt-get update && apt-get install -y --no-install-recommends \
ca-certificates \
build-essential \
git \
python \
python-pip \
python-setuptools
RUN pip install tf-nightly-gpu
# Checkout tensorflow/models and pin to a particular version
RUN git clone https://github.com/tensorflow/models.git /tensorflow_models && \
cd /tensorflow_models && git checkout feat/multi-gpu-resnet
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment