File size: 933 Bytes
3da5127
 
 
 
 
 
 
59a743a
 
2385631
59a743a
31ade5b
59a743a
b330d38
 
 
3da5127
59a743a
3da5127
59a743a
 
2385631
31ade5b
b330d38
3da5127
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
#!/bin/sh
set -eu

if [ -z "${OMP_NUM_THREADS:-}" ]; then
    export OMP_NUM_THREADS=1
fi

export CUDA_VISIBLE_DEVICES=-1
export NVIDIA_VISIBLE_DEVICES=void
export ROBOMME_RENDER_BACKEND="${ROBOMME_RENDER_BACKEND:-pci:0}"
unset NVIDIA_DRIVER_CAPABILITIES
unset SAPIEN_RENDER_DEVICE
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] ROBOMME_RENDER_BACKEND=$ROBOMME_RENDER_BACKEND"
echo "[entrypoint] SAPIEN_RENDER_DEVICE=${SAPIEN_RENDER_DEVICE:-<unset>}"
echo "[entrypoint] VK_ICD_FILENAMES=${VK_ICD_FILENAMES:-<unset>}"
exec "$@"