Mirrors/photoprism
1
0
Fork 0
mirror of https://github.com/photoprism/photoprism.git synced 2026-01-23 02:24:24 +00:00
Code Issues Projects Releases Wiki Activity

AI: Detect AMX-AVX512 instructions in install-tensorflow.sh #222 #4922

Browse source 
Signed-off-by: Michael Mayer <michael@photoprism.app>
This commit is contained in:
Michael Mayer 2025-04-06 10:01:22 +02:00
parent 5650ea29b5
commit a3456637cd
1 changed files with 8 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

10
scripts/dist/install-tensorflow.sh vendored
View file

@ -66,9 +66,15 @@ if [[ $TF_DRIVER == "auto" ]]; then
CPU_DETECTED=$(lshw -c processor -json 2>/dev/null)
if [[ $(echo "${CPU_DETECTED}" | jq -r '.[0].capabilities.avx_vnni') == "true" ]]; then
if [[ $(echo "${CPU_DETECTED}" | jq -r '.[0].capabilities.amx_tile') == "true" ]]; then
TF_DRIVER="avx512"
echo "✅ Your CPU supports AMX-AVX512 Advanced Matrix Extensions"
if [[ $(lshw -c display -json 2>/dev/null | jq -r '.[0].configuration.driver') == "nvidia" ]]; then
echo "✅ You can optionally install the tensorflow-gpu version as an Nvidia GPU was detected"
fi
elif [[ $(echo "${CPU_DETECTED}" | jq -r '.[0].capabilities.avx_vnni') == "true" ]]; then
TF_DRIVER="vnni"
echo "✅ Your CPU supports AVX-VNNI vector neural network instructions"
echo "✅ Your CPU supports AVX-VNNI Vector Neural Network Instructions"
if [[ $(lshw -c display -json 2>/dev/null | jq -r '.[0].configuration.driver') == "nvidia" ]]; then
echo "✅ You can optionally install the tensorflow-gpu version as an Nvidia GPU was detected"
fi