Skip to content

TensorFlow module

Erik Smistad edited this page Dec 13, 2020 · 7 revisions

To enable this module set FAST_MODULE_TensorFlow in CMake, reconfigure and compile. A couple of new CMake options will appear (FAST_BUILD_TensorFlow_*), one for each version of TensorFlow, by default only the TensorFlow CPU version is built. You can build the CUDA and ROCm versions by selecting these in CMake. Please make sure you have the requirements installed before you proceed to build TensorFlow (see below). With these options Tensorflow will be downloaded from github, configured and built automatically when you build FAST. Note that Tensorflow is a big library and compiling it takes a long time, especially on Windows.

Current TensorFlow version used by FAST is 1.14.
TensorFlow will from FAST version 3.2 be using TensorFlow 2.3.0.

Examples using the TensorFlow module

Requirements

Clone this wiki locally