Skip to content

Commit

Permalink
Try minimal dockerfile for build workflow debugging
Browse files Browse the repository at this point in the history
  • Loading branch information
Scott Davidson committed Sep 22, 2023
1 parent f749954 commit 863a420
Showing 1 changed file with 7 additions and 10 deletions.
17 changes: 7 additions & 10 deletions images/pytorch-benchmarks/Dockerfile
Original file line number Diff line number Diff line change
@@ -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/
# COPY run-benchmark.sh /usr/local/bin/

0 comments on commit 863a420

Please sign in to comment.