TensorFlow: Update install-tensorflow.sh script

Signed-off-by: Michael Mayer <michael@photoprism.app>
This commit is contained in:
Michael Mayer 2023-04-29 11:51:06 +02:00
parent 199d3ef2af
commit 1f31426515

View file

@ -57,6 +57,8 @@ if [[ $TF_DRIVER == "auto" ]]; then
TF_DRIVER="avx2"
elif [[ $(echo "${CPU_DETECTED}" | jq -r '.[].capabilities.avx') == "true" ]]; then
TF_DRIVER="avx"
else
TF_DRIVER=""
fi
fi