diff --git a/images/pytorch-benchmarks/Dockerfile b/images/pytorch-benchmarks/Dockerfile index d7d79ca..02a36bd 100644 --- a/images/pytorch-benchmarks/Dockerfile +++ b/images/pytorch-benchmarks/Dockerfile @@ -7,7 +7,6 @@ WORKDIR /workspace/benchmark # Add other benchmarks here? RUN python install.py alexnet resnet50 llama -# PyTorch install.py pins numpy=1.21.2 but this breaks numba so update both here -RUN pip install -U numpy numba - -COPY run-benchmark.sh /usr/local/bin/ \ No newline at end of file +# PyTorch install.py pins numpy=1.21.2 but +# this breaks numba so update both here +RUN pip install -U numpy numba \ No newline at end of file