stable robomme easiest one step

Dockerfile

@@ -13,6 +13,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     libgl1 \
     libglib2.0-0 \
     libvulkan1 \
+    mesa-vulkan-drivers \
     && add-apt-repository ppa:deadsnakes/ppa \
     && apt-get update && apt-get install -y --no-install-recommends \
     python3.11 \

docker-entrypoint.sh

@@ -9,12 +9,15 @@ export CUDA_VISIBLE_DEVICES=-1
 export NVIDIA_VISIBLE_DEVICES=void
 export SAPIEN_RENDER_DEVICE=cpu
 unset NVIDIA_DRIVER_CAPABILITIES
-unset VK_ICD_FILENAMES
 unset MUJOCO_GL
+if [ -z "${VK_ICD_FILENAMES:-}" ] && [ -f /usr/share/vulkan/icd.d/lvp_icd.x86_64.json ]; then
+    export VK_ICD_FILENAMES=/usr/share/vulkan/icd.d/lvp_icd.x86_64.json
+fi
 
 echo "[entrypoint] Starting RoboMME Gradio app in CPU-only mode"
 echo "[entrypoint] OMP_NUM_THREADS=$OMP_NUM_THREADS"
 echo "[entrypoint] CUDA_VISIBLE_DEVICES=$CUDA_VISIBLE_DEVICES"
 echo "[entrypoint] NVIDIA_VISIBLE_DEVICES=$NVIDIA_VISIBLE_DEVICES"
 echo "[entrypoint] SAPIEN_RENDER_DEVICE=$SAPIEN_RENDER_DEVICE"
+echo "[entrypoint] VK_ICD_FILENAMES=${VK_ICD_FILENAMES:-<unset>}"
 exec "$@"

{scripts → gradio-web}/minimal_robomme_env_cpu_step.py

@@ -28,8 +28,11 @@ def configure_cpu_only_runtime() -> None:
     os.environ["NVIDIA_VISIBLE_DEVICES"] = "void"
     os.environ["SAPIEN_RENDER_DEVICE"] = "cpu"
     os.environ.pop("NVIDIA_DRIVER_CAPABILITIES", None)
-    os.environ.pop("VK_ICD_FILENAMES", None)
     os.environ.pop("MUJOCO_GL", None)
+    if "VK_ICD_FILENAMES" not in os.environ:
+        lvp_icd = Path("/usr/share/vulkan/icd.d/lvp_icd.x86_64.json")
+        if lvp_icd.exists():
+            os.environ["VK_ICD_FILENAMES"] = str(lvp_icd)
 
 
 configure_cpu_only_runtime()