From f27abcb265e6e8a29271af2b9e243fa68c8af465 Mon Sep 17 00:00:00 2001 From: Andrey Arapov Date: Fri, 26 Jan 2024 23:23:39 +0100 Subject: [PATCH] fix benchmark test example manifest fixes #52 --- example/pod/alexnet-gpu.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/example/pod/alexnet-gpu.yaml b/example/pod/alexnet-gpu.yaml index cba6f36d..552ca012 100644 --- a/example/pod/alexnet-gpu.yaml +++ b/example/pod/alexnet-gpu.yaml @@ -13,7 +13,7 @@ spec: - 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: ["python3 benchmarks/scripts/tf_cnn_benchmarks/tf_cnn_benchmarks.py --model=alexnet; 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