Yin-Shane-Xia / tensorflow_wrapper

Tensorflow Installation Wrapper

A small project using CMake to build tensorflow C++ from source (Google's github tagged releases) on x86_64 and aarch64 (specifically arm64v8) with GPU support.

Build Requirements

You can easily package these components into a Dockerfile if you want to go that route.

If you are using a TX2 or Xavier, just be sure to package your nvidia, cuda, and cudnn drivers/libs into the image as well. Also, in the utils folder there is a script that creates the scratch space necessary to build tensorflow on the TX2/Xavier target. The script contains the repo link if you want more information on how that works.

Prerequisites

Bazel

Grab Bazel 0.15.0 for tensorflow 1.11. If you are building on aarch64, you will need to build bazel from source on the target device or setup bazel to build a cross-compiled binary (not-recommended).

RUN mkdir -p /opt/bazel && cd /opt/bazel && \
	wget --quiet https://github.com/bazelbuild/bazel/releases/download/0.15.0/bazel-0.15.0-dist.zip && \
	unzip bazel-0.15.0-dist.zip && rm bazel-0.15.0-dist.zip && \
	cd /opt/bazel && bash ./compile.sh && mv ./output/bazel /usr/local/bin && rm -rf /opt/bazel

Protobuf

Install protobuf 3.6.0 (required by tensorflow).

RUN cd /opt && \
	wget --quiet https://github.com/protocolbuffers/protobuf/releases/download/v3.6.0/protobuf-all-3.6.0.tar.gz && \
	tar -zxf protobuf-all-3.6.0.tar.gz && rm protobuf-all-3.6.0.tar.gz && \
	cd protobuf-3.6.0 && \
	./autogen.sh && ./configure && \
	make && make install && ldconfig && \
	rm -rf /opt/protobuf-3.6.0

Eigen

Install Eigen 3

RUN cd /opt && \
	wget --quiet https://bitbucket.org/eigen/eigen/get/fd6845384b86.tar.gz && \
	tar -zxf fd6845384b86.tar.gz && rm /opt/fd6845384b86.tar.gz && \
	mkdir -p /opt/eigen-eigen-fd6845384b86/build && cd /opt/eigen-eigen-fd6845384b86/build && \
	cmake .. && \
	make && make install && \
	rm -rf /opt/eigen-eigen-fd6845384b86 && \
	ln -s /usr/local/include/eigen3 /usr/include/eigen3

GCC-6

RUN apt-get update && apt-get install -y --no-install-recommends software-properties-common && \
	add-apt-repository ppa:ubuntu-toolchain-r/test -y && \
	apt-get update && apt-get install -y --no-install-recommends gcc-6 g++-6 && \
	update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-6 60 --slave /usr/bin/g++ g++ /usr/bin/g++-6

Tensorflow

RUN cd /opt && \
	git clone https://github.com/cding/tensorflow_wrapper.git && \
	mkdir -p /opt/tensorflow_wrapper/build && \
	cd /opt/tensorflow_wrapper/build && \
	cmake .. && make && make install

On TX2 and Xavier, Make sure your /usr/lib/aarch64-linux-gnu/libcuda.so.1 exists and is linked to /usr/lib/aarch64-linux-gnu/libcuda.so otherwise the build will fail.