Mirror
/
LocalAI
mirror of https://github.com/mudler/LocalAI.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

feat(entrypoint): detect GPU 

Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
This commit is contained in:
Ettore Di Giacinto 2024-03-02 09:44:58 +01:00
parent 1c312685aa
commit 360b355397
1 changed files with 28 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
entrypoint.sh
View File

@ -47,4 +47,32 @@ else
echo "@@@@@"
fi
set +e
GPU_ACCELERATION=false
case "$(uname -s)" in
Linux)
if lspci | grep -E 'VGA|3D' | grep -iq nvidia; then
echo "NVIDIA GPU detected"
GPU_ACCELERATION=true
elif lspci | grep -E 'VGA|3D' | grep -iq amd; then
echo "AMD GPU detected"
GPU_ACCELERATION=true
elif lspci | grep -E 'VGA|3D' | grep -iq intel; then
echo "Intel GPU detected"
GPU_ACCELERATION=true
fi
;;
Darwin)
if system_profiler SPDisplaysDataType | grep -iq 'Metal'; then
echo "Apple Metal supported GPU detected"
GPU_ACCELERATION=true
fi
;;
esac
set -e
if [ "$GPU_ACCELERATION" = "false" ]; then
echo "!!! No GPU detected, running with CPU only !!!"
fi
./local-ai "$@"