Skip to content

modernizing Dockerfile syntax #154

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
azrael417 wants to merge 6 commits into
base: main
Choose a base branch
from
Open

modernizing Dockerfile syntax #154

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
5f0f0a1
modernizing Dockerfile syntax
azrael417 Jul 16, 2026
bfb46c6
Modify TORCH_CUDA_ARCH_LIST in Dockerfile
azrael417 Jul 16, 2026
706eca0
Modify TORCH_CUDA_ARCH_LIST in Dockerfile.aws
azrael417 Jul 16, 2026
a4018f9
making onnxruntime gpu optional
azrael417 Jul 16, 2026
2c395a2
Merge branch 'tkurth/dockerfile-fix' of https://github.com/NVIDIA/mak…
azrael417 Jul 16, 2026
19a7b71
removing dummy env
azrael417 Jul 16, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 5 additions & 4 deletions docker/Dockerfile
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,8 @@ RUN apt remove cmake -y && \
RUN pip install mpi4py

# install onnx
RUN pip install onnxruntime onnxruntime-gpu
RUN pip install onnxruntime && \
if [ "$(uname -m)" = "x86_64" ]; then pip install onnxruntime-gpu; fi

# hdf5 and h5py
ENV HDF5_VERSION=2.0.0
Expand Down Expand Up @@ -74,8 +75,8 @@ RUN pip install git+https://github.com/NVIDIA/physicsnemo.git@v1.3.0

# torch-harmonics
RUN pip install setuptools_scm
ENV FORCE_CUDA_EXTENSION 1
ENV TORCH_CUDA_ARCH_LIST "8.0 8.6 8.7 9.0 10.0+PTX"
ENV FORCE_CUDA_EXTENSION=1
ENV TORCH_CUDA_ARCH_LIST="8.0 8.6 8.7 9.0a 10.0a+PTX"
RUN cd /opt && git clone https://github.com/NVIDIA/torch-harmonics.git && \
cd torch-harmonics && \
pip install --no-build-isolation -e .
Expand All @@ -90,4 +91,4 @@ COPY makani /opt/makani/makani
COPY tests /opt/makani/tests
COPY pyproject.toml /opt/makani/pyproject.toml
COPY README.md /opt/makani/README.md
RUN cd /opt/makani && pip install --no-deps -e .
RUN cd /opt/makani && pip install --no-deps -e .
12 changes: 6 additions & 6 deletions docker/Dockerfile.aws
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,8 @@ RUN apt remove cmake -y && \
RUN pip install mpi4py

# install onnx
RUN pip install onnxruntime onnxruntime-gpu
RUN pip install onnxruntime && \
if [ "$(uname -m)" = "x86_64" ]; then pip install onnxruntime-gpu; fi

# install AWS prereqs for HDF5
RUN cd /opt && git clone https://github.com/awslabs/aws-c-common.git && \
Expand Down Expand Up @@ -107,7 +108,7 @@ RUN cd /opt && git clone https://github.com/awslabs/aws-c-auth.git && \
make install

# 7. aws-c-s3 (final target for HDF5)
RUn cd /opt && git clone https://github.com/awslabs/aws-c-s3.git && \
RUN cd /opt && git clone https://github.com/awslabs/aws-c-s3.git && \
cd aws-c-s3 && \
mkdir build && cd build && \
cmake .. -DCMAKE_BUILD_TYPE=Release && \
Expand Down Expand Up @@ -156,9 +157,8 @@ RUN pip install "git+https://github.com/google-research/weatherbench2.git"
RUN pip install --ignore-installed "git+https://github.com/NVIDIA/mlperf-common.git"

# torch-harmonics
ENV FORCE_CUDA_EXTENSION 1
ENV TORCH_CUDA_ARCH_LIST "8.0 8.6 8.7 9.0 10.0 12.0+PTX"
ENV HARMONICS_VERSION 0.8.5
ENV FORCE_CUDA_EXTENSION=1
ENV TORCH_CUDA_ARCH_LIST="8.0 8.6 8.7 9.0a 10.0a 12.0+PTX"
RUN cd /opt && git clone https://github.com/NVIDIA/torch-harmonics.git && \
cd torch-harmonics && \
pip install --no-build-isolation -e .
Expand All @@ -176,4 +176,4 @@ COPY makani /opt/makani/makani
COPY tests /opt/makani/tests
COPY pyproject.toml /opt/makani/pyproject.toml
COPY README.md /opt/makani/README.md
RUN cd /opt/makani && pip install -e .
RUN cd /opt/makani && pip install -e .
Loading