From 36897fc0e98c27728bb43e172269a82766e0c9cd Mon Sep 17 00:00:00 2001 From: "fri.K" Date: Tue, 19 Nov 2024 14:01:33 +0100 Subject: [PATCH] Cannot initialize CUDA even if nvidia-smi detected CUDA can't be properly initialized if cuda-compat on host and docker mismatches. Removing cuda-compat-11-2 package fixes the issue Issue discussion: https://community.opendronemap.org/t/opendronemap-nodeodm-gpu-nvidia-smi-detected-cannot-initialize-cuda/21124 [INFO] nvidia-smi detected [INFO] Using CPU for feature extraction: Cannot initialize CUDA --- gpu.Dockerfile | 1 + 1 file changed, 1 insertion(+) diff --git a/gpu.Dockerfile b/gpu.Dockerfile index dae7f189..e6e1c47b 100644 --- a/gpu.Dockerfile +++ b/gpu.Dockerfile @@ -38,6 +38,7 @@ 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/lib/x86_64-linux-gnu/libavcodec.so.58 /usr/lib/x86_64-linux-gnu/libavcodec.so.58 +RUN dpkg --remove cuda-compat-11-2 RUN apt-get update -y \ && apt-get install -y ffmpeg libtbb2 # Install shared libraries that we depend on via APT, but *not*