-
Notifications
You must be signed in to change notification settings - Fork 20
Expand file tree
/
Copy pathsetup.py
More file actions
executable file
·24 lines (21 loc) · 831 Bytes
/
setup.py
File metadata and controls
executable file
·24 lines (21 loc) · 831 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
import time, os, sys, shutil
import custom_build_tools.build as build_tools
pwd = sys.path[0]
# build TF
print('[INFO] Building Tensorflow...')
print("[INFO] It's going to last for about 90 minutes on Intel i7-6700")
# time.sleep(1)
python_exec = build_tools.get_python_path()
python_bin = '/'.join(build_tools.get_python_path().split('/')[:-1])
python_lib_path = build_tools.get_python_package()
build_tools.config(python_exec, python_lib_path)
build_tools.build_bazel()
print('[INFO] Build Complete')
# copy true setup.py into install dir
shutil.copyfile(pwd + '/.setup.py', pwd + '/tensorflow/tools/pip_package/setup.py')
print('[INFO] Building wheel')
build_tools.build_pip()
print('[INFO] Wheel Build Complete\n')
print('[INFO] Installing wheel')
build_tools.install_pip(python_bin + '/pip')
print('[INFO] Instal Complete')