From 5c132846a2bba3303d75abc2c51f6bc373e596e0 Mon Sep 17 00:00:00 2001
From: Anthony Musco <anthony.g.musco@gmail.com>
Date: Tue, 13 Feb 2024 17:59:25 -0700
Subject: [PATCH] Install vainfo and mesa-va-drivers

When building for docker, the OpenGL renderer context needs to be opened
with the same driver that the point clouds are generated in (e.g. NVIDIA).
Prior to this change, the OpenGL renderer driver defaulted to LLVM and thus
caused a CUDA check error when the GUI attempted to bind a buffer.

This change also adds mesa-utils, which isn't 100% necessary but is useful
for debugging future graphics issues.

Closes #95
---
 Dockerfile | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/Dockerfile b/Dockerfile
index f83e0b9..0c80f31 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -28,7 +28,15 @@ RUN apt update && apt-get -qq install -y --no-install-recommends \
         libqt5x11extras5-dev\
         libqt5opengl5-dev   \
         libsuitesparse-dev  \
-        meshlab
+        meshlab             \
+        mesa-utils          \
+        vainfo              \
+        mesa-va-drivers
+
+# nvidia-container-runtime
+ENV NVIDIA_VISIBLE_DEVICES ${NVIDIA_VISIBLE_DEVICES:-all}
+ENV NVIDIA_DRIVER_CAPABILITIES ${NVIDIA_DRIVER_CAPABILITIES:+$NVIDIA_DRIVER_CAPABILITIES,}graphics
+ENV MESA_D3D12_DEFAULT_ADAPTER_NAME=NVIDIA
 
 RUN wget https://www.python.org/ftp/python/3.9.9/Python-3.9.9.tgz
 RUN tar xzvf Python-3.9.9.tgz