Skip to content
Snippets Groups Projects
Commit 2a20c44e authored by Jeff Poznanovic's avatar Jeff Poznanovic Committed by Poznanovic, Jeffrey
Browse files

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.
parent 719507a4
No related tags found
No related merge requests found
......@@ -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
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment