Skip to content

Tensorflow 1.7 GPU can be compiled in Windows and used in TensorflowSharp #252

Open
@edwardyehuang

Description

@edwardyehuang

Just let you guys know. I have succeeded compiled the Tensorflow 1.7 GPU version, and also successfully used in TensorflowSharp with the large graph model (Before the Tensorflow GPU 1.5, the size of the pb graph is limited at 64mb)

In order to compile the lastest Tensorflow dll, you need to use the CMake:
https://github.com/tensorflow/tensorflow/tree/master/tensorflow/contrib/cmake

Be sure enable the GPU support and shared library support when using the CMake to generate the project files

Be sure using the Visual Studio 2015 to compile. The Cuda version is 9.0 and cudnn version is the lastest.

Be sure to change " tf_core_kernels.vcxproj" properties:
Properties->Configuration Properties->VC++ Directories->Executable Directories
$(VC_ExecutablePath_x64) to $(VC_ExecutablePath_x64_x64)

After compiling, make sure you added your cudnn/bin folder to "PATH". And rename the tensorflow.dll to libtensorflow.dll

Metadata

Metadata

Assignees

No one assigned

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions