Skip to content

Latest commit

 

History

History
28 lines (22 loc) · 1.02 KB

README.md

File metadata and controls

28 lines (22 loc) · 1.02 KB
Raw

Extract headers from TensorFlow compiling directory

For some C++ headers needed by TFRA are not included in installation directory of old version Tensorflow. If you want to compile TFRA with these TF, it is necessary for extracting headers from TensorFlow compiling dirctory. Do as below:

  1. Compile TensorFlow with version you need to work with

  1. Modify BAZEL_CACHE_DIR in ./env.sh, for $BAZEL_CACHE_DIR is created randomly, so change it after step 1 : _BAZEL_CACHE_DIR=$BAZEL_CACHE_DIR/_bazel_root/860b4776177384b48cc611bdf8c7a91c

  2. Extract headers by executing ./mkinclude.sh

./mkinclude.sh
  1. Copy $OUTPUT_INCLUDE_DIR to TFRA compile direcory
cp -r $_BAZEL_CACHE_DIR $TFRA_HOME/build_deps/tf_header/$TF_VERSION/
  1. Start compiling TFRA
PY_VERSION="3.7" TF_VERSION="1.15.2" TF_NEED_CUDA=0 sh .github/workflows/make_wheel_Linux.sh