TensorFlow: Update install-tensorflow.sh script
Signed-off-by: Michael Mayer <michael@photoprism.app>
This commit is contained in:
parent
199d3ef2af
commit
1f31426515
1 changed files with 2 additions and 0 deletions
2
scripts/dist/install-tensorflow.sh
vendored
2
scripts/dist/install-tensorflow.sh
vendored
|
@ -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
|
||||
|
||||
|
|
Loading…
Add table
Reference in a new issue