This project aims to give a tutorial training a MNIST model in python, reading it in java, verifying that you can use it in both languages with the same result.
To run this example you need the following.
You also need to download and unpack the mnist training data. The gzip archives could be found at THE MNIST DATABASE. Unpacking these and adding them to a folder called mnist so the scripts can find the files. In windows the unpacked files could get extensions that make them unrecognized. Removing these extensions so the ending filenames are in the format of mnist/t10k-labels.idx1-ubyte
As of me writing this the current version of tensorflow is 1.0.0-PREVIEW1, this version is missing the feature SavedModelBundle which makes the setup much easier. By running only one save you get a dir with a full model. So you could modify this demo to use the preview but building tensorflow in Linux or Mac are not that hard.
Tensorflow uses the build system bazel, a install description could be found at Building bazel or following the steps below. (Tested on ubuntu trusty)
sudo add-apt-repository ppa:webupd8team/java
sudo apt-get update
sudo apt-get install oracle-java8-installer
sudo apt-get install pkg-config zip g++ zlib1g-dev unzip
echo "deb [arch=amd64] http://storage.googleapis.com/bazel-apt stable jdk1.8" | sudo tee /etc/apt/sources.list.d/bazel.list
curl https://bazel.build/bazel-release.pub.gpg | sudo apt-key add -
sudo apt-get update && sudo apt-get install bazel
sudo apt-get upgrade bazel
The easiest way to install tensorflow is with a pip package. To build your own package and install follow instructions at Building tenserflow or the description below.
git clone https://github.com/tensorflow/tensorflow.git
cd tensorflow
sudo pip install six numpy wheel
bazel build --config=opt //tensorflow/tools/pip_package:build_pip_package
sudo pip install /tmp/tensorflow_pkg/tensorflow-1.0.1-py2-none-any.whl
Last but not least we need the java bindings, these can be built by following the description at Building tenserflow jar or the operations below.
If you haven't done so already you need to fetch the source from github in order to build java bindings. Follow instructions below.
git clone https://github.com/tensorflow/tensorflow.git
cd tensorflow
sudo apt-get install python swig python-numpy
./configure
bazel build --config opt //tensorflow/java:tensorflow //tensorflow/java:libtensorflow_jni
The JAR (libtensorflow.jar) and native library (libtensorflow_jni.so) will be in bazel-bin/tensorflow/java.
Copy the .so file to the jni directory in this project so ./run.sh can find it. And install the jar in maven your repository by following the description below.
After we built the bindings we can install them in our local maven repository with
mvn org.apache.maven.plugins:maven-install-plugin:2.5.2:install-file
-Dfile=libtensorflow.jar
-DgroupId=org.tensorflow
-DartifactId=libtensorflow
-Dversion=1.1.0-MINE
-Dpackaging=jar
-DgeneratePom=true
Training can be done with python mnist_train.py in the project directory.
Verifying with java we can build the project with mvn package and run it with ./run.sh in the project directory. This script will run the project jar with the jni binding library.
Remove visual studio installations https://github.com/Microsoft/VisualStudioUninstaller
Download python and install with the "Add python to environment variables" option. https://www.python.org/
Download and install msys2. http://www.msys2.org/
Download and install bazel distribution from the link below. https://github.com/bazelbuild/bazel/releases
Download cuda and cudnn for GPU support. https://developer.nvidia.com/cuda-downloads https://developer.nvidia.com/cudnn
cd [bazel-dist-dir]
pacman -Syuu --noconfirm
pacman -Syuu gcc git curl zip unzip zlib-devel --noconfirm
export BAZEL_VS=c:/Program\ Files\ \(x86\)/Microsoft\ Visual\ Studio\ 14.0
export BAZEL_PYTHON=c:/tools/python36/python.exe
export BAZEL_SH=c:/tools/msys64/usr/bin/bash.exe
export PATH=$PATH:/c/tools/python36
./compile.sh
export BAZEL_VS=c:/Program\ Files\ (x86)/Microsoft\ Visual\ Studio\ 12.0 export BAZEL_VS=c:/Program\ Files\ (x86)/Microsoft\ Visual\ Studio/2017/Community export BAZEL_VS=c:/Program\ Files\ (x86)/Microsoft\ Visual\ Studio/2017/BuildTools
export BAZEL_VS=c:/Program\ Files\ \(x86\)/Microsoft\ Visual\ Studio\ 14.0
export BAZEL_PYTHON=c:/tools/python36/python.exe
export BAZEL_SH=c:/tools/msys64/usr/bin/bash.exe
export PATH=$PATH:/c/tools/python36
git clone https://github.com/tensorflow/tensorflow.git
cd [tensorflow]
pacman -Syuu patch --noconfirm
pip install six numpy wheel protobuf
export PYTHON_BIN_PATH=c:/tools/python36/python.exe
export PYTHON_LIB_PATH=c:/tools/python36/lib/site-packages
export PYTHON_INCLUDE_PATH=c:/tools/python36/include
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:c:/cuda/extras/CUPTI/libWin32
export CUDA_PATH=c:/cuda
export CUDA_PATH_V8_0=c:/cuda
export CUDNN_INSTALL_PATH=c:/cuda
export PATH=$PATH:/c/cuda/bin:/c/github/bazel-0.12.0/output/
$ ./configure
Extracting Bazel installation...
You have bazel 0.8.0- (@non-git) installed.
Do you wish to build TensorFlow with XLA JIT support? [y/N]:
No XLA JIT support will be enabled for TensorFlow.
Do you wish to build TensorFlow with GDR support? [y/N]:
No GDR support will be enabled for TensorFlow.
Do you wish to build TensorFlow with VERBS support? [y/N]:
No VERBS support will be enabled for TensorFlow.
Do you wish to build TensorFlow with CUDA support? [y/N]: y
CUDA support will be enabled for TensorFlow.
Please specify the CUDA SDK version you want to use, e.g. 7.0. [Leave empty to default to CUDA 8.0]: 9.0
Please specify the location where CUDA 9.0 toolkit is installed. Refer to README.md for more details. [Default is c:/cuda]:
Please specify the cuDNN version you want to use. [Leave empty to default to cuDNN 6.0]: 7.0.4
Please specify a list of comma-separated Cuda compute capabilities you want to build with.
You can find the compute capability of your device at: https://developer.nvidia.com/cuda-gpus.
Please note that each additional compute capability significantly increases your build time and binary size. [Default is: 3.5,5.2]6.1
Do you wish to build TensorFlow with MPI support? [y/N]:
No MPI support will be enabled for TensorFlow.
Please specify optimization flags to use during compilation when bazel option "--config=opt" is specified [Default is -march=native]:
Add "--config=mkl" to your bazel command to build with MKL support.
Please note that MKL on MacOS or windows is still not supported.
If you would like to use a local MKL instead of downloading, please set the environment variable "TF_MKL_ROOT" every time before build.
Configuration finished
bazel build -c opt --jobs 1 tensorflow/tools/pip_package:build_pip_package
bazel build //tensorflow/tools/pip_package:build_pip_package
bazel build -c opt tensorflow/tools/pip_package:build_pip_package
bazel build --config=monolithic //tensorflow/tools/pip_package:build_pip_package
bazel build --action_env=USE_MSVC_WRAPPER=1 //tensorflow/tools/pip_package:build_pip_package
Copy all files from
cp --preserve=links -r bazel-tensorflow bazel-bin/tensorflow/tools/pip_package/build_pip_package.runfiles/
Create PIP package
bazel-bin/tensorflow/tools/pip_package/build_pip_package c:/tmp/tensorflow_pkg
pip install c:\tmp\tensorflow_pkg\tensorflow-1.1.0rc1-cp35-cp35m-win_amd64.whl
bazel build -c opt $BUILD_OPTS //tensorflow/java:tensorflow //tensorflow/java:libtensorflow_jni
bazel build --config opt //tensorflow/java:tensorflow //tensorflow/java:libtensorflow_jni
cd bazel-bin\tensorflow\java\
copy libtensorflow.jar to your build directory
copy libtensorflow_jni.so to tensorflow_jni.dll in your jni directory.
Download
Install python pip packages
pip install six numpy wheel protobuf
Prepare build directory
git clone https://github.com/tensorflow/tensorflow.git
cd tensorflow
git checkout v1.8.0
cd tensorflow\tensorflow\contrib\cmake
mkdir build
cd build
Configure for Python 3.6
cmake .. -A x64 -DCMAKE_BUILD_TYPE=Release -DSWIG_EXECUTABLE=C:/tools/swigwin-3.0.12/swig.exe -DPYTHON_EXECUTABLE=c:/tools/Python36/python.exe -DPYTHON_LIBRARIES=c:/tools/Python36/libs/python36.lib -Dtensorflow_ENABLE_GPU=ON -DCUDNN_HOME=c:/cuda -Dtensorflow_VERBOSE=ON
Build example trainer
MSBuild /maxcpucount:2 /p:Configuration=Release tf_tutorials_example_trainer.vcxproj
Build pip packages
MSBuild /maxcpucount:2 /p:Configuration=Release tf_python_build_pip_package.vcxproj
export BAZEL_WRKDIR=c:/tempdir/shrtpath
export BAZEL_SH=c:/tools/msys64/usr/bin/bash.exe
export BAZEL_VS=c:/Program\ Files\ \(x86\)/Microsoft\ Visual\ Studio\ 14.0
export BAZEL_PYTHON=c:/tools/Python27/python.exe
export EXTRA_BAZEL_ARGS='--crosstool_top=@standalone_local_config_cc//:toolchain'
export EXTRA_BAZEL_ARGS='--copt=-Ic:/tools/msys64/usr/include --copt=-Lc:/tools/msys64/usr/lib --verbose_failures'
export EXTRA_BAZEL_ARGS='--jobs 1'
export CFLAGS="-Ic:/tools/msys64/usr/include"
export CXXFLAGS="-Ic:/tools/msys64/usr/include"
export LDFLAGS="-Lc:/tools/msys64/usr/lib"
export EXTRA_BAZEL_ARGS='--jobs 1 --cpu=x64_windows_msvc --host_cpu=x64_windows_msvc --verbose_failures'
export BAZEL_VS=c:/Program\ Files\ \(x86\)/Microsoft\ Visual\ Studio\ 14.0
export EXTRA_BAZEL_ARGS='--cpu=x64_windows_msvc --host_cpu=x64_windows_msvc --copt=-w --host_copt=-w --python_path=c:/tools/Python27/python.exe --verbose_failures'
export BAZEL_WRKDIR=c:/tempdir/shrtpath
export BAZEL_SH=c:/tools/msys64/usr/bin/bash.exe
export BAZEL_VS=c:/Program\ Files\ \(x86\)/Microsoft\ Visual\ Studio\ 14.0
export BAZEL_PYTHON=c:/tools/Python36/python.exe
export PATH=$PATH:/c/cuda/bin:/c/tools/Python36:/c/github/bazel-0.5.4/output/
export BUILD_OPTS='--cpu=x64_windows_msvc --host_cpu=x64_windows_msvc --python_path=c:/tools/python27/python.exe --verbose_failures'
export BUILD_OPTS='--cpu=x64_windows_msvc --host_cpu=x64_windows_msvc --copt=-w --host_copt=-w --python_path=c:/tools/python36/python.exe --verbose_failures'
export BUILD_OPTS='--cpu=x64_windows_msvc --host_cpu=x64_windows_msvc --verbose_failures'
bazel build --config=opt --config=cuda --cpu=x64_windows_msvc --copt=-w --host_copt=-w --host_cpu=x64_windows_msvc --verbose_failures //tensorflow/tools/pip_package:build_pip_package
bazel build -c opt $BUILD_OPTS tensorflow/tools/pip_package:build_pip_package
bazel build --config=opt --config=cuda //tensorflow/tools/pip_package:build_pip_package
./compile.sh compile output/bazel.exe
export BAZEL_VS=c:/Program\ Files\ \(x86\)/Microsoft\ Visual\ Studio/2017/BuildTools
export EXTRA_BAZEL_ARGS='--cpu=x64_windows_msvc --host_cpu=x64_windows_msvc --copt=-w --host_copt=-w --python_path=c:/tools/python36/python.exe --verbose_failures'