Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- apiVersion: "kubeflow.org/v1alpha2"
- kind: "TFJob"
- metadata:
- name: "tf-smoke-gpu"
- spec:
- tfReplicaSpecs:
- PS:
- replicas: 1
- template:
- metadata:
- creationTimestamp: null
- spec:
- containers:
- - args:
- - python
- - tf_cnn_benchmarks.py
- - --batch_size=32
- - --model=resnet50
- - --variable_update=parameter_server
- - --flush_stdout=true
- - --num_gpus=1
- - --local_parameter_device=cpu
- - --device=cpu
- - --data_format=NHWC
- image: gcr.io/kubeflow/tf-benchmarks-cpu:v20171202-bdab599-dirty-284af3
- name: tensorflow
- ports:
- - containerPort: 2222
- name: tfjob-port
- resources:
- limits:
- cpu: '1'
- workingDir: /opt/tf-benchmarks/scripts/tf_cnn_benchmarks
- restartPolicy: OnFailure
- Worker:
- replicas: 1
- template:
- metadata:
- creationTimestamp: null
- spec:
- containers:
- - args:
- - python
- - tf_cnn_benchmarks.py
- - --batch_size=32
- - --model=resnet50
- - --variable_update=parameter_server
- - --flush_stdout=true
- - --num_gpus=1
- - --local_parameter_device=cpu
- - --device=gpu
- - --data_format=NHWC
- image: gcr.io/kubeflow/tf-benchmarks-gpu:v20171202-bdab599-dirty-284af3
- name: tensorflow
- ports:
- - containerPort: 2222
- name: tfjob-port
- resources:
- limits:
- nvidia.com/gpu: 1
- workingDir: /opt/tf-benchmarks/scripts/tf_cnn_benchmarks
- restartPolicy: OnFailure
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement