Skip to content

Commit

Permalink
try building deepcam from mlcommons (#38)
Browse files Browse the repository at this point in the history
* try building deepcam from mlcommons

Signed-off-by: vsoch <vsoch@users.noreply.github.com>
  • Loading branch information
vsoch committed Apr 12, 2023
1 parent d94dc7b commit 01b4c95
Show file tree
Hide file tree
Showing 2 changed files with 31 additions and 0 deletions.
24 changes: 24 additions & 0 deletions mlcommons-deepcam/Dockerfile
Original file line number Diff line number Diff line change
@@ -0,0 +1,24 @@
ARG tag="21.12-py3"
FROM nvcr.io/nvidia/pytorch:${tag}
ARG commit=ca8af13c2b6996fe60622406b1c6826a3cd34a7c

# some requirements
RUN pip install h5py==3.2.1

#install mpi4py
RUN pip install mpi4py==3.0.3

#install mlperf logging
RUN pip install "git+https://github.com/mlperf/logging.git@501bbde47f005d67c6357da6e5c1931eab339f8e"

# core assets
RUN git clone https://github.com/mlcommons/hpc /tmp/hpc && \
cd /tmp/hpc && \
git checkout ${commit} && \
cd ./deepcam && \
mv ./src/deepCam /opt/deepCam && \
mv ./src/utils /opt/utils

# additional folders for mapping data
RUN mkdir -p /data

7 changes: 7 additions & 0 deletions mlcommons-deepcam/uptodate.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,7 @@
dockerbuild:

build_args:
tag:
key: tag
versions:
- "21.12-py3"

0 comments on commit 01b4c95

Please sign in to comment.