fo40225 / tensorflow-windows-wheel

Tensorflow prebuilt binary for Windows
3.66k stars 1.53k forks source link

need custom build #75

Open largeword opened 5 years ago

largeword commented 5 years ago

hi there, can you build tensorflow 1.13 for avx512 , cuda10.1 and tensorrt5.1? i want to know how fast can it be at win10 when use tensorrt5.1 but i dont know how to build it. thank you so much

fo40225 commented 5 years ago

@largeword Tensorflow didn't support TensorRT on Windows.

The best way to use avx512 on windows is icl but I don't think bazel on windows support other compiler than msvc.

largeword commented 5 years ago

@largeword Tensorflow didn't support TensorRT on Windows.

The best way to use avx512 on windows is icl but I don't think bazel on windows support other compiler than msvc.

ok, thanks, but i dont know how to do that, can you write an article or something else?

fo40225 commented 5 years ago

A long time ago, I had build the tensorflow with Intel compiler by cmake.

I haven't try it in the most recent version.

cd tensorflow/contrib/cmake
md build
cd build

cmake .. -G "Visual Studio 15 2017" -A x64 -T "Intel C++ Compiler 19.0",host=x64 ^
-DCMAKE_BUILD_TYPE=Release ^
-DSWIG_EXECUTABLE=C:\Users\User\swigwin-3.0.10\swig.exe ^
-Dtensorflow_BUILD_CC_EXAMPLE=OFF ^
-Dtensorflow_BUILD_SHARED_LIB=ON ^
-Dtensorflow_BUILD_PYTHON_TESTS=ON ^
-Dtensorflow_WIN_CPU_SIMD_OPTIONS="/QxCORE-AVX512"

cmake --build . --target tf_python_build_pip_package --config Release -- /m /fileLogger
pip install tf_python\dist\tensorflow.whl
fo40225 commented 5 years ago

@largeword

You can use Intel Optimization for TensorFlow instead of build it yourself.

https://software.intel.com/en-us/articles/intel-optimization-for-tensorflow-installation-guide

largeword commented 5 years ago

@largeword

You can use Intel Optimization for TensorFlow instead of build it yourself.

https://software.intel.com/en-us/articles/intel-optimization-for-tensorflow-installation-guide

thank you so much, i'll have a try