Compile Tensorflow: Difference between revisions
No edit summary |
No edit summary |
||
Line 3: | Line 3: | ||
Verify dependencies | Verify dependencies | ||
<pre>sudo apt install python3-dev python3-pip</pre> | <pre>sudo apt install python3-dev python3-pip</pre> | ||
Install Dependencies | |||
<pre>sudo pip3 install numpy</pre> | |||
Install Bazel using Bazelisk | Install Bazel using Bazelisk |
Revision as of 22:54, 26 October 2021
Tensorflow packages often assume the processor supports avx2. Our Xeon process does not, and requires tensorflow to be compiled without this flag.
Verify dependencies
sudo apt install python3-dev python3-pip
Install Dependencies
sudo pip3 install numpy
Install Bazel using Bazelisk
sudo npm install -g @bazel/bazelisk
Checkout Tensorflow
git clone https://github.com/tensorflow/tensorflow.git cd tensorflow
Configure
Keep all defaults
./configure
Compile
The jobs parameter may be needed to reduce memory dependencies on some servers
bazel build --jobs 2 //tensorflow/tools/pip_package:build_pip_package
I got compilation errors at one point and some guidance said to add the following to the command to reduce memory consumption:
--local_ram_resources 2048
or
bazel build --jobs 2 --local_ram_resources 2048 //tensorflow/tools/pip_package:build_pip_package