Is it possible to use TensorFlow C++ API on Windows?

前端 未结 5 1661
滥情空心
滥情空心 2020-12-14 03:49

I\'m interested in incorporating TensorFlow into a C++ server application built in Visual Studio on Windows 10 and I need to know if that\'s possible. Google recently annou

相关标签:
5条回答
  • 2020-12-14 04:08

    There is a detailed guide by Joe Antognini and a similar TensorFlow ReadMe at GitHub explaining the building of TensorFlow source via CMake. You also need to have SWIG installed on your machine which allows connecting C/C++ source with the Python scripting language. I did use Visual CMAKE (cmake-gui) with the screen capture shown below.

    In the CMake configuration, I used Visual Studio 15 2017 compiler. Once this stage successfully completes, you can click on the Generate button to go ahead with the actual build process.

    However, on Visual Studio 2015, when I attempted building via the "ALL_BUILD" project, the setup gave me "build tools for v141 cannot be found" error. This did not go away even when I attempted to retarget my solution. Finally, the solution got built successfully with Visual Studio 2017. You also need to manually set the SWIG_EXECUTABLE path in CMake before it successfully configures.

    As indicated in the Antognini link, for me the build took about half an hour on a 16GB RAM, Core i7 machine. Once done, you might want to validate your build by attempting to run the tf_tutorials_example_trainer.exe file.

    Hope this helps!

    0 讨论(0)
  • 2020-12-14 04:13

    I had to use a downgraded version of my Visual Studio 2017 (from 15.7.5 to 15.4) by adding "VC++ 2017 version 15.4 v14.11 toolset" through the installer (Individual Components tab).

    The cmake command which worked for me was:

    cmake .. -A x64 -DCMAKE_BUILD_TYPE=Release ^
    -T "v141,version=14.11" ^
    -DSWIG_EXECUTABLE="C:/Program Files/swigwin-3.0.12/swig.exe" ^
    -DPYTHON_EXECUTABLE="C:/Program Files/Python/python.exe" ^
    -DPYTHON_LIBRARIES="C:/Program Files/Python/libs/python27.lib" ^
    -Dtensorflow_ENABLE_GPU=ON ^
    -DCUDNN_HOME="C:/Program Files/cudnn-9.2-windows10-x64-v7.1/cuda"  ^
    -DCUDA_TOOLKIT_ROOT_DIR="C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v9.0"
    

    After the build, open tensorflow.sln in Visual Studio and build ALL_BUILD.

    If you want to enable GPU computation, do check your Graphics Card here (Compute Capability > 3.5). Do remember to install all the packages (Cuda Toolkit 9.0, cuDNN, Python 3.7, SWIG, Git, CMake...) and add the paths to the environment variable in the beginning.

    0 讨论(0)
  • 2020-12-14 04:13

    I made a README detailing how to I built the Tensorflow dll and .lib file for the C++ API on Windows with GPU support building from source with Bazel. Tensorflow version 1.14

    The tutorial is step by step and starts at the very beginning, so you may have to scroll down past steps you have already done, like checking your hardware, installing Bazel etc. Here is the url: https://github.com/sitting-duck/stuff/tree/master/ai/tensorflow/build_tensorflow_1.14_source_for_Windows

    Probably you will want to scroll all the way down to this part: https://github.com/sitting-duck/stuff/tree/master/ai/tensorflow/build_tensorflow_1.14_source_for_Windows#step-7-build-the-dll

    It shows how to pass command to create .lib and .dll.

    Then to test your .lib you should link it into your c++ project,

    Then it will show you how to identify and fix the missing symbols using the TF_EXPORT macro

    I am actively working on making this tutorial better so feel free to leave comments on this answer if you are having problems.

    0 讨论(0)
  • It is certainly possible to use TensorFlow's C++ API on Windows, but it is not currently very easy. Right now, the easiest way to build against the C++ API on Windows would be to build with CMake, and adapt the CMake rules for the tf_tutorials_example_trainer project (see the source code here). Building with CMake will give you a Visual Studio project in which you can implement your C++ TensorFlow program.

    Note that the tf_tutorials_example_trainer project builds a Console Application that statically links all of the TensorFlow runtime into your program. At present we have not written the necessary rules to create a reusable TensorFlow DLL, although this would be technially possible: for example, the Python extension is a DLL that includes the runtime, but does not export the necessary symbols to use TensorFlow's C or C++ APIs directly.

    0 讨论(0)
  • 2020-12-14 04:26

    For our latest work on building TensorFlow C++ API on Windows, please look at this github page. This works on Windows 10, currently without CUDA support (only CPU).

    PS: Only the bazel build method works, because CMake is not supported and not maintained anymore, resulting in CMake configuration errors.

    0 讨论(0)
提交回复
热议问题