Update Dockerfile

Dockerfile

@@ -31,12 +31,21 @@ RUN apt-get update && apt-get install --no-install-recommends -y \
     libgles2-mesa-dev \
     cmake \
     curl
+    
+ENV PYTHONDONTWRITEBYTECODE=1
+ENV PYTHONUNBUFFERED=1
+
+# for GLEW
+ENV LD_LIBRARY_PATH /usr/lib64:$LD_LIBRARY_PATH
+
+# nvidia-container-runtime
+ENV NVIDIA_VISIBLE_DEVICES all
+ENV NVIDIA_DRIVER_CAPABILITIES compute,utility,graphics
+
+# Default pyopengl to EGL for good headless rendering support
+ENV PYOPENGL_PLATFORM egl
 
-# Set EGL-related environment variables
-ENV LD_LIBRARY_PATH=/usr/lib64:$LD_LIBRARY_PATH \
-    NVIDIA_VISIBLE_DEVICES=all \
-    NVIDIA_DRIVER_CAPABILITIES=compute,utility,graphics \
-    PYOPENGL_PLATFORM=egl
+COPY ./10_nvidia.json /usr/share/glvnd/egl_vendor.d/10_nvidia.json
 
 # Set up a new user named "user" with user ID 1000
 RUN useradd -m -u 1000 user