Install time command in container

stackhpc · Sep 25, 2023 · 1e083a8 · 1e083a8
1 parent 83a2ee0
commit 1e083a8
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 4 deletions.
diff --git a/images/pytorch-benchmarks/Dockerfile b/images/pytorch-benchmarks/Dockerfile
@@ -1,6 +1,6 @@
 FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
 
-RUN apt update && apt install -y git
+RUN apt update && apt install -y git time
 # TODO: Checkout certain commit here
 RUN git clone https://github.com/pytorch/benchmark
 WORKDIR /workspace/benchmark

diff --git a/images/pytorch-benchmarks/run-benchmark.sh b/images/pytorch-benchmarks/run-benchmark.sh