no libtensorflow for arm

livepeer · Oct 9, 2023 · a5b469a · a5b469a
1 parent 179e102
commit a5b469a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/install_ffmpeg.sh b/install_ffmpeg.sh
@@ -184,13 +184,13 @@ elif [[ "$GOOS" == "windows" ]]; then
 elif [[ -e "/usr/local/cuda/lib64" ]]; then
   echo "CUDA SDK detected, building with GPU support"
   EXTRA_FFMPEG_FLAGS="$EXTRA_FFMPEG_FLAGS --enable-nonfree --enable-cuda-nvcc --enable-libnpp --enable-cuda --enable-cuda-llvm --enable-cuvid --enable-nvenc --enable-decoder=h264_cuvid,hevc_cuvid,vp8_cuvid,vp9_cuvid --enable-filter=scale_npp,signature_cuda,hwupload_cuda --enable-encoder=h264_nvenc,hevc_nvenc"
-  if [[ ! -e "${ROOT}/compiled/lib/libtensorflow_framework.so" ]]; then
+  if [[ "$GOARCH" == "amd64" && ! -e "${ROOT}/compiled/lib/libtensorflow_framework.so" ]]; then
     LIBTENSORFLOW_VERSION=2.12.1 &&
       curl -LO https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-gpu-linux-x86_64-${LIBTENSORFLOW_VERSION}.tar.gz &&
       tar -C ${ROOT}/compiled/ -xzf libtensorflow-gpu-linux-x86_64-${LIBTENSORFLOW_VERSION}.tar.gz &&
       rm libtensorflow-gpu-linux-x86_64-${LIBTENSORFLOW_VERSION}.tar.gz
+    EXTRA_FFMPEG_FLAGS="$EXTRA_FFMPEG_FLAGS --enable-libtensorflow"
   fi
-  EXTRA_FFMPEG_FLAGS="$EXTRA_FFMPEG_FLAGS --enable-libtensorflow"
 else
   echo "No CUDA SDK detected, building without GPU support"
 fi