Sfoglia il codice sorgente

Update photoprism-install.sh

Use regex to determine if CPU has avx capabilities
If no avx capabilities detected swith to cpu tensorflow automatically
aurelpec 3 anni fa
parent
commit
68830d02e3
1 ha cambiato i file con 7 aggiunte e 22 eliminazioni
  1. 7 22
      setup/photoprism-install.sh

+ 7 - 22
setup/photoprism-install.sh

@@ -103,29 +103,14 @@ chown root:root /usr/local/sbin/gosu
 chmod 755 /usr/local/sbin/gosu
 msg_ok "Installed Golang"
 
-read -p "Check For Advanced Vector Extensions? " -n 1 -r
-echo
-if [[ $REPLY =~ ^[Yy]$ ]]
-then
-    check=yes
-else
-    check=no
-fi
-
 msg_info "Installing Tensorflow"
-if [ "$check" == "yes" ]; then 
-  if [[ "$AVX" == "avx2" ]]; then
-    wget https://dl.photoprism.org/tensorflow/linux/libtensorflow-linux-avx2-1.15.2.tar.gz &>/dev/null
-    tar -C /usr/local -xzf libtensorflow-linux-avx2-1.15.2.tar.gz &>/dev/null
-    elif [[ "$AVX" == "avx" ]]; then
-    wget https://dl.photoprism.org/tensorflow/linux/libtensorflow-linux-avx-1.15.2.tar.gz &>/dev/null
-    tar -C /usr/local -xzf libtensorflow-linux-avx-1.15.2.tar.gz &>/dev/null
-    elif [[ "$AVX" == " " ]]; then
-    wget https://dl.photoprism.org/tensorflow/linux/libtensorflow-linux-cpu-1.15.2.tar.gz &>/dev/null
-    tar -C /usr/local -xzf libtensorflow-linux-cpu-1.15.2.tar.gz &>/dev/null
-  fi
-fi
-if [ "$check" == "no" ]; then 
+if [[ "$AVX" =~ avx2 ]]; then
+  wget https://dl.photoprism.org/tensorflow/linux/libtensorflow-linux-avx2-1.15.2.tar.gz &>/dev/null
+  tar -C /usr/local -xzf libtensorflow-linux-avx2-1.15.2.tar.gz &>/dev/null
+  elif [[ "$AVX" =~ avx ]]; then
+  wget https://dl.photoprism.org/tensorflow/linux/libtensorflow-linux-avx-1.15.2.tar.gz &>/dev/null
+  tar -C /usr/local -xzf libtensorflow-linux-avx-1.15.2.tar.gz &>/dev/null
+  else
   wget https://dl.photoprism.org/tensorflow/linux/libtensorflow-linux-cpu-1.15.2.tar.gz &>/dev/null
   tar -C /usr/local -xzf libtensorflow-linux-cpu-1.15.2.tar.gz &>/dev/null
 fi