Blame view
tools/extras/install_tensorflow_cc.sh
1.71 KB
8dcb6dfcb first commit |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 |
#!/bin/bash set -e #export JAVA_HOME=/LOCATION_ON_YOUR_MACHINE/java/jdk1.8.0_121 PATH=$PWD/bazel/output:$PATH export HOME=$PWD/tensorflow_build/ mkdir -p $HOME java=`which java` if [ -z "$JAVA_HOME" ]; then java=$JAVA_HOME/bin/java fi java_version=`$java -version 2>&1 | head -n 1 | awk '{print $3}' | sed s=\"==g` good_version=`echo 1.8 $java_version | awk '{if($1<$2)print 1; else print 0}'` if [ $good_version -eq 0 ]; then echo You have jdk version = $java_version, which is older than 1.8 echo You need to download a later than 1.8 JDK version at echo http://www.oracle.com/technetwork/pt/java/javase/downloads/jdk8-downloads-2133151.html echo and set your JAVA_HOME to point to where it is installed exit 1 else echo You have jdk version = $java_version, which is newer than 1.8. We will continue the installation fi [ ! -f bazel.zip ] && wget https://github.com/bazelbuild/bazel/releases/download/0.15.0/bazel-0.15.0-dist.zip -O bazel.zip mkdir -p bazel cd bazel unzip ../bazel.zip ./compile.sh cd ../ # now bazel is built [ ! -d tensorflow ] && git clone https://github.com/tensorflow/tensorflow cd tensorflow git fetch --tags git checkout r1.12 ./configure tensorflow/contrib/makefile/download_dependencies.sh bazel build -c opt //tensorflow:libtensorflow.so bazel build -c opt //tensorflow:libtensorflow_cc.so echo Building tensorflow completed. You will need to go to kaldi/src/ and do echo \"make\" under tfrnnlm/ and tfrnnlmbin/ to generate the binaries exit # the following would utilize the highest optimization but might not work in a # grid where each machine might have different configurations bazel build --config=opt //tensorflow:libtensorflow.so bazel build --config=opt //tensorflow:libtensorflow_cc.so |