From 863a4209c6f129b53f8fda0bc8a62323250bca95 Mon Sep 17 00:00:00 2001 From: Scott Davidson Date: Fri, 22 Sep 2023 16:54:26 +0100 Subject: [PATCH] Try minimal dockerfile for build workflow debugging --- images/pytorch-benchmarks/Dockerfile | 17 +++++++---------- 1 file changed, 7 insertions(+), 10 deletions(-) diff --git a/images/pytorch-benchmarks/Dockerfile b/images/pytorch-benchmarks/Dockerfile index 72fe8c4..48e708c 100644 --- a/images/pytorch-benchmarks/Dockerfile +++ b/images/pytorch-benchmarks/Dockerfile @@ -1,13 +1,10 @@ -# TODO: Try newer versions of base image - 23.08 fails with -# 'CUDA on host is too old' type error but something in between -# should work. Latest 23.08 might also work on Ubuntu-22.04. FROM nvcr.io/nvidia/pytorch:22.12-py3 -RUN pip install -U pip -RUN pip install torchvision torchaudio -RUN git clone https://github.com/pytorch/benchmark -WORKDIR /workspace/benchmark -# Add other benchmarks here? -RUN python install.py alexnet resnet50 llama +# RUN pip install -U pip +# RUN pip install torchvision torchaudio +# RUN git clone https://github.com/pytorch/benchmark +# WORKDIR /workspace/benchmark +# # Add other benchmarks here? +# RUN python install.py alexnet resnet50 llama -COPY run-benchmark.sh /usr/local/bin/ \ No newline at end of file +# COPY run-benchmark.sh /usr/local/bin/ \ No newline at end of file