Cannot build Open3D with CUDA in Docker

Accelerated Computing NGC GPU Cloud Docker and NVIDIA Docker
,
1

Hello everybody !

First of all, a big thanks for the people who will spend some times for answering to this post.
I am currently trying to create a docker container with CUDA enable for a specific application with an IFM 3D camera (which run on Ubuntu 18.04 with CUDA 10.2 and Nvidia runtime).
The application is running in CPU mode, but to enable the GPU (CUDA) we need to build the Open3D library from source.

To do that, I have tested two containers, one from “l4t-cuda:10.2.460-runtime” and the second one from “l4t-base:r32.4.3”.
Both finish with the same error code during CMake process :
Unable to find cudart library.
CMake Error with the function FindCUDAToolkit => Could NOT find CUDAToolkit (missing: CUDA_CUDART) (found version “10.2.300”).
I really don’t know what can I do and why I have this error message with the base container from Nvidia directly :(

229786793-35013f70-d793-495d-95b9-9c2c82212507
229786793-35013f70-d793-495d-95b9-9c2c822125071112×494 16.5 KB

My Dockerfile :

FROM nvcr.io/nvidia/l4t-cuda:10.2.460-runtime

USER root

# Installation de GIT
RUN apt-get update && \
    apt-get install -y git sudo && \
    apt-get clean

# Installation de CMAKE
RUN apt update && \
    apt install -y software-properties-common && \
    apt clean all && \
    wget -O - https://apt.kitware.com/keys/kitware-archive-latest.asc 2>/dev/null | gpg --dearmor - | tee /etc/apt/trusted.gpg.d/kitware.gpg >/dev/null && \
    apt-add-repository "deb https://apt.kitware.com/ubuntu/ bionic main" && \
    apt update && \
    apt install -y cmake

# Téléchargement et installation de Open3D
ARG OPEN3D_CLONE_REPOSITORY="https://github.com/isl-org/Open3D"
ARG OPEN3D_VERSION="v0.17.0"
RUN git clone --depth 1 --branch ${OPEN3D_VERSION} ${OPEN3D_CLONE_REPOSITORY} && \
    cd Open3D && \
    util/install_deps_ubuntu.sh assume-yes && \
    mkdir build && \
    cd build && \
    cmake -DCUDAToolkit_ROOT=/usr/local/cuda-10.2/lib -DCMAKE_INSTALL_PREFIX=/usr \
        -DBUILD_CUDA_MODULE=ON \
        -DBUILD_PYTORCH_OPS=OFF \
        -DBUILD_TENSORFLOW_OPS=OFF \
        -DPYTHON_EXECUTABLE=/usr/bin/python3 \
        -DBUILD_PYTHON_MODULE=ON \
        -DBUILD_GUI=OFF \
        -DBUNDLE_OPEN3D_ML=OFF \
        -DOPEN3D_ML_ROOT=https://github.com/isl-org/Open3D-ML.git \
        .. && \
    make -j$(nproc) && \
    make install && \
    make -j install-pip-package