diff --git a/example/pod/alexnet-gpu.yaml b/example/pod/alexnet-gpu.yaml index cba6f36da156db69355def848bfbc75544fc7920..552ca0121092511d65f3cadc1a18745de3afe7fe 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