Skip to content

Instantly share code, notes, and snippets.

FROM python:3.6.4-slim-stretch
RUN apt-get update && apt-get install -y build-essential git time
RUN pip install https://anaconda.org/intel/tensorflow/1.6.0/download/tensorflow-1.6.0-cp36-cp36m-linux_x86_64.whl
RUN git clone https://github.com/tensorflow/models.git
ENV PYTHONPATH $PYTHONPATH:$HOME/models
WORKDIR $HOME/models/official/resnet/
RUN pip3 install -r ../requirements.txt
apiVersion: v1
kind: Pod
metadata:
name: stress-pod
spec:
containers:
- image: gcr.io/google-containers/stress:v1
args: ["-cpus", "48"]
name: stress-container
resources:
apiVersion: v1
kind: Pod
metadata:
name: tf-wide-deep-census-pod
spec:
containers:
- image: <insert-image>
command: ["/bin/sh"]
args: ["-c", "/usr/bin/time python ./wide_deep.py --model_type=wide_deep --train_epochs=500 --epochs_between_eval=500 --batch_size=32561"]
name: tf-wide-deep-census
FROM python:3.6-slim-stretch
RUN apt-get update && apt-get install -y git time
RUN pip install https://anaconda.org/intel/tensorflow/1.6.0/download/tensorflow-1.6.0-cp36-cp36m-linux_x86_64.whl
RUN git clone https://github.com/tensorflow/models.git
ENV PYTHONPATH $PYTHONPATH:$HOME/models
WORKDIR $HOME/models/official/wide_deep
RUN python ./data_download.py
apiVersion: v1
kind: Pod
metadata:
name: parsec-pod
spec:
containers:
# See https://github.com/Spirals-Team/benchmark-containers/tree/master/parsec-3.0 for more information on image.
- image: spirals/parsec-3.0
command: ["/bin/bash"]
# Change args to run the workload.
apiVersion: v1
kind: Pod
metadata:
name: tf-wide-deep-census-pod
spec:
containers:
- image: <insert-image>
command: ["/bin/sh"]
args: ["-c", "/usr/bin/time python ./wide_deep.py --model_type=wide_deep --train_epochs=500 --epochs_between_eval=500 --batch_size=32561"]
name: tf-wide-deep-census