Skip to content

Commit

Permalink
update with main branch modifications
Browse files Browse the repository at this point in the history
  • Loading branch information
pablospe committed Feb 6, 2024
1 parent 9c88596 commit 1ffab03
Showing 1 changed file with 36 additions and 4 deletions.
40 changes: 36 additions & 4 deletions Dockerfile
Original file line number Diff line number Diff line change
@@ -1,6 +1,5 @@
ARG UBUNTU_VERSION=22.04
ARG NVIDIA_CUDA_VERSION=12.3.1
FROM nvidia/cuda:${NVIDIA_CUDA_VERSION}-devel-ubuntu${UBUNTU_VERSION} as builder
FROM ubuntu:${UBUNTU_VERSION} as builder

ENV QT_XCB_GL_INTEGRATION=xcb_egl

Expand Down Expand Up @@ -36,7 +35,40 @@ RUN apt-get install -y --no-install-recommends --no-install-suggests wget && \
-DCMAKE_INSTALL_PREFIX=/ceres_installed && \
ninja install
RUN cp -r /ceres_installed/* /usr/local/

# Build pyceres.
ADD . /pyceres
COPY . /pyceres
WORKDIR /pyceres
RUN pip install . -vv
RUN pip install --upgrade pip
RUN pip wheel . --no-deps -w dist-wheel -vv && \
whl_path=$(find dist-wheel/ -name "*.whl") && \
echo $whl_path >dist-wheel/whl_path.txt


#
# Runtime stage.
#
FROM ubuntu:${UBUNTU_VERSION} as runtime

# Install minimal runtime dependencies.
RUN apt-get update && \
apt-get install -y --no-install-recommends --no-install-suggests \
libgoogle-glog0v5 \
libspqr2 \
libcxsparse3 \
libatlas3-base \
python-is-python3 \
python3-minimal \
python3-pip

# Copy installed library in the builder stage.
COPY --from=builder /ceres_installed/ /usr/local/

# Install pyceres.
COPY --from=builder /pyceres/dist-wheel /tmp/dist-wheel
RUN pip install --upgrade pip
RUN cd /tmp && whl_path=$(cat dist-wheel/whl_path.txt) && pip install $whl_path
RUN rm -rfv /tmp/*

# # Verify if pyceres library is accessible from python.
RUN python -c "import pyceres"

0 comments on commit 1ffab03

Please sign in to comment.