working gpu docker build

pull/1774/head
Evan Hallein 2024-07-04 21:39:11 +08:00
rodzic 8864fcdd03
commit 709afd16f3
1 zmienionych plików z 9 dodań i 9 usunięć

Wyświetl plik

@ -22,7 +22,7 @@ RUN bash configure.sh clean
### Use a second image for the final asset to reduce the number and
# size of the layers.
#FROM nvidia/cuda:11.2.2-runtime-ubuntu20.04
FROM nvidia/cuda:11.2.0-devel-ubuntu20.04
#FROM nvidia/cuda:11.2.0-devel-ubuntu20.04
# Env variables
ENV DEBIAN_FRONTEND=noninteractive \
@ -30,21 +30,21 @@ ENV DEBIAN_FRONTEND=noninteractive \
LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/code/SuperBuild/install/lib" \
PDAL_DRIVER_PATH="/code/SuperBuild/install/bin"
WORKDIR /code
#WORKDIR /code
# Copy everything we built from the builder
COPY --from=builder /code /code
#COPY --from=builder /code /code
# Copy the Python libraries installed via pip from the builder
COPY --from=builder /usr/local /usr/local
#COPY --from=builder /usr/local /usr/local
# Install shared libraries that we depend on via APT, but *not*
# the -dev packages to save space!
# Also run a smoke test on ODM and OpenSfM
RUN bash configure.sh installruntimedepsonly \
&& apt-get clean \
&& rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/* \
&& bash run.sh --help \
&& bash -c "eval $(python3 /code/opendm/context.py) && python3 -c 'from opensfm import io, pymap'"
#RUN bash configure.sh installruntimedepsonly \
# && apt-get clean \
# && rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/* \
# && bash run.sh --help \
# && bash -c "eval $(python3 /code/opendm/context.py) && python3 -c 'from opensfm import io, pymap'"
# Entry point
ENTRYPOINT ["python3", "/code/run.py"]