Skip to content

Reduce container image size #749

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

Merged
yousefmoazzam merged 2 commits into from
Aug 14, 2026
+5 −11
Merged

Reduce container image size #749

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
2eab9d4
Reduce container image size by removing unnecessary deps
yousefmoazzam Aug 14, 2026
0acd13f
Add OpenMPI env var to Dockerfile
yousefmoazzam Aug 14, 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
16 changes: 5 additions & 11 deletions Dockerfile
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,23 +1,17 @@
FROM mambaorg/micromamba:2.0.8-debian12-slim AS build

RUN micromamba install -y python=3.12 blas[build=mkl] 'numpy=2.4.*' scipy tqdm pyyaml pillow click gcc
RUN micromamba install -y openmpi=4.1.6 mpi
RUN micromamba install -y h5py[build=*openmpi*] hdf5plugin
RUN micromamba install -y mpi4py
RUN micromamba install -y aiofiles graypy imageio loguru pluggy pytest iniconfig
RUN micromamba install -y tomopy=1.15.3 scikit-image
RUN micromamba install -y cuda-version=12.9 cuda-cudart==12.9.* cupy=14.0.* nvtx
RUN micromamba install -y -c astra-toolbox -c conda-forge astra-toolbox::astra-toolbox
RUN micromamba run python -m pip install tomobar
RUN micromamba run python -m pip install --no-deps httomolib httomolibgpu httomo-backends
RUN micromamba install -y -c astra-toolbox -c conda-forge python=3.12 blas[build=mkl] 'numpy=2.4.*' tqdm pyyaml pillow click gcc openmpi=4.1.6 mpi h5py[build=*openmpi*] hdf5plugin mpi4py aiofiles graypy imageio loguru iniconfig tomopy=1.15.3 scikit-image cuda-version=12.9 cuda-cudart==12.9.* cupy=14.0.* nvtx astra-toolbox::astra-toolbox
RUN micromamba run python -m pip install --no-cache-dir --no-deps tomobar httomolib httomolibgpu httomo-backends
RUN micromamba uninstall -n base -y gcc
RUN micromamba clean -y --all --force-pkgs-dirs

COPY . .
RUN micromamba run python -m pip install --no-deps .
RUN micromamba run python -m pip install --no-cache-dir --no-deps .

FROM nvidia/cuda:12.9.1-base-ubuntu24.04 AS deploy

COPY --from=build /opt/conda /opt/conda
ENV CONDA_PREFIX=/opt/conda
ENV OMPI_MCA_plm_rsh_agent=

ENTRYPOINT ["/opt/conda/bin/python", "-m", "httomo", "run"]
Loading