Merge pull request #1813 from fri-K/master

Cannot initialize CUDA even if nvidia-smi detected
OpenDroneMap · Jan 13, 2025 · 75b9b92 · 75b9b92
2 parents df0c585 + 36897fc
commit 75b9b92
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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*