-
Notifications
You must be signed in to change notification settings - Fork 3
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Try minimal dockerfile for build workflow debugging
- Loading branch information
Scott Davidson
committed
Sep 22, 2023
1 parent
f749954
commit 863a420
Showing
1 changed file
with
7 additions
and
10 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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/ |