From a5b469a8865fc80b0c0ad4a6f1ef6407929d4eec Mon Sep 17 00:00:00 2001 From: Eli Mallon Date: Mon, 9 Oct 2023 16:47:52 -0700 Subject: [PATCH] no libtensorflow for arm --- install_ffmpeg.sh | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/install_ffmpeg.sh b/install_ffmpeg.sh index 27c5afc67f..8bf673c2a9 100755 --- 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