photoprism/docker/tensorflow/Makefile

21 lines
888 B
Makefile
Raw Normal View History

2020-04-27 20:22:55 +02:00
TF_VERSION=1.15.2
# -march see https://gcc.gnu.org/onlinedocs/gcc-4.8.5/gcc/i386-and-x86-64-Options.html
all: libtensorflow libtensorflow-avx libtensorflow-avx2
patch:
git apply tensorflow-$(TF_VERSION).diff
2020-04-28 11:51:42 +02:00
download:
wget https://github.com/tensorflow/tensorflow/archive/v$(TF_VERSION).tar.gz
tar -xzf v$(TF_VERSION).tar.gz
cp .tf_configure.bazelrc Makefile *.sh tensorflow-$(TF_VERSION)
libtensorflow:
bazel build --jobs 2 --config=opt //tensorflow:libtensorflow.so
./create_archive.sh linux-cpu $(TF_VERSION)
libtensorflow-avx:
bazel build --jobs 2 --config=opt //tensorflow:libtensorflow.so --copt=-march=core-avx-i --host_copt=-march=core-avx-i
./create_archive.sh linux-avx $(TF_VERSION)
libtensorflow-avx2:
bazel build --jobs 2 --config=opt //tensorflow:libtensorflow.so --copt=-march=core-avx2 --host_copt=-march=core-avx2
./create_archive.sh linux-avx2 $(TF_VERSION)