Correct structure

This commit is contained in:
Alexandre
2025-01-08 12:50:27 +01:00
committed by GitHub
parent 58f20101a8
commit 29b7b3a2a2

View File

@@ -2,11 +2,19 @@
# shellcheck shell=bash
set -e
# Compensate for old cpu without avx2
if [[ "$(uname -m)" = "x86_64" && lscpu | grep -q "Flags" && ! lscpu | grep -q "avx2" ]]; then
bashio::log.warning "NON SUPPORTED CPU DETECTED"
bashio::log.warning "Your cpu doesn't support avx2, the analyzer service will likely won't work"
bashio::log.warning "Trying to install tensorflow instead of tflite_runtime instead"
$PYTHON_VIRTUAL_ENV /usr/bin/pip3 uninstall -y tflite_runtime
$PYTHON_VIRTUAL_ENV /usr/bin/pip3 install tensorflow
# Check if the CPU supports AVX2
if [[ "$(uname -m)" = "x86_64" ]]; then
# Get the CPU flags
cpu_flags=$(lscpu | grep "Flags" | awk '{print $2}')
# Check if avx2 is NOT present in the flags
if [[ ! "$cpu_flags" =~ "avx2" ]]; then
bashio::log.warning "NON SUPPORTED CPU DETECTED"
bashio::log.warning "Your CPU doesn't support AVX2, the analyzer service likely won't work."
bashio::log.warning "Trying to install tensorflow instead of tflite_runtime."
# Uninstall tflite_runtime and install tensorflow
$PYTHON_VIRTUAL_ENV /usr/bin/pip3 uninstall -y tflite_runtime
$PYTHON_VIRTUAL_ENV /usr/bin/pip3 install tensorflow
fi
fi