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

This commit is contained in:
thelamer 2025-08-08 09:18:28 -04:00
parent bd75fc04ab
commit c91d542012
No known key found for this signature in database

View File

@ -1,7 +1,7 @@
#!/bin/bash #!/bin/bash
# Enable Nvidia GPU support if detected # 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 LIBGL_KOPPER_DRI2=1
export MESA_LOADER_DRIVER_OVERRIDE=zink export MESA_LOADER_DRIVER_OVERRIDE=zink
export GALLIUM_DRIVER=zink export GALLIUM_DRIVER=zink