From 2a20c44e25715f7dba9e659e1aa2c747889952d1 Mon Sep 17 00:00:00 2001
From: Jeff Poznanovic <jeffrey.poznanovic@amd.com>
Date: Fri, 19 Apr 2019 16:50:15 -0600
Subject: [PATCH] Update pod deployment example yaml
The existing alexnet-gpu.yaml file contained an out-of-date docker
image (rocm/tensorflow:rocm1.7.1). It is being updated to use
rocm/tensorflow:latest instead.
Also, the example AlexNet workload command can be improved by using
the official tensorflow/benchmarks project. This replaces the use
of the unofficial convnet-benchmarks.
---
example/pod/alexnet-gpu.yaml | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/example/pod/alexnet-gpu.yaml b/example/pod/alexnet-gpu.yaml
index 37fb96b2..f1c2519c 100644
--- a/example/pod/alexnet-gpu.yaml
+++ b/example/pod/alexnet-gpu.yaml
@@ -7,13 +7,13 @@ metadata:
spec:
containers:
- name: alexnet-tf-gpu-container
- image: rocm/tensorflow:rocm1.7.1
+ image: rocm/tensorflow:latest
workingDir: /root
env:
- name: HIP_VISIBLE_DEVICES
value: "0" # # 0,1,2,...,n for running on GPU and select the GPUs, -1 for running on CPU
command: ["/bin/bash", "-c", "--"]
- args: ["python convnet-benchmarks/tensorflow/benchmark_alexnet.py; trap : TERM INT; sleep infinity & wait"]
+ args: ["python3 benchmarks/scripts/tf_cnn_benchmarks/tf_cnn_benchmarks.py --model=alexnet; trap : TERM INT; sleep infinity & wait"]
resources:
limits:
amd.com/gpu: 1 # requesting a GPU
--
GitLab