photoprism/docker/scripts/install-tensorflow.sh
2021-12-12 20:15:52 +01:00

25 lines
846 B
Bash
Executable file

#!/usr/bin/env bash
if [[ -z $1 ]] && [[ -z $2 ]]; then
echo "Please define architecture and variant (cpu/avx2/...)" 1>&2
exit 1
else
echo "Downloading tensorflow library..."
if [[ $1 == "amd64" ]]; then
TARGETARCH="linux"
TARGETVARIANT="-${2:-"cpu"}"
URL="https://dl.photoprism.app/tensorflow/$TARGETARCH/libtensorflow-${TARGETARCH}${TARGETVARIANT}-1.15.2.tar.gz"
elif [[ $1 == "arm64" ]]; then
URL="https://dl.photoprism.app/tensorflow/$1/libtensorflow-$1-1.15.2.tar.gz"
elif [[ $1 == "arm" ]]; then
URL="https://dl.photoprism.app/tensorflow/$1/libtensorflow-$1-1.15.2.tar.gz"
else
echo "cpu architecture not supported by now" 1>&2
exit 1
fi
echo "$URL"
curl -fsSL "$URL" | \
tar --overwrite -C "/usr" -xz && \
ldconfig
echo "Done"
fi