update nvidia detection to handle hosts that use runtime with no GPU

This commit is contained in:
thelamer 2025-08-08 09:26:14 -04:00
parent 0ee73a1122
commit a5383bbc61
No known key found for this signature in database

View File

@ -1,7 +1,7 @@
#!/usr/bin/with-contenv bash
# Enable Nvidia GPU support if detected
if which nvidia-smi; then
if which nvidia-smi > /dev/null 2>&1 && ls -A /dev/dri 2>/dev/null && [ "${DISABLE_ZINK}" == "false" ]; then
export LIBGL_KOPPER_DRI2=1
export MESA_LOADER_DRIVER_OVERRIDE=zink
export GALLIUM_DRIVER=zink