PatWie/tensorflow-cmake

Building dll with cuda 10

ttdd11 opened this issue · 2 comments

Great repo!

Have you tried building tensorflow_cc.dll with gpu options?

There is no windows support yet as I cannot test it and I never had the need to do so.

But the tests are using CUDA10 on Linux

- name: inference_cc
pull: never
image: tensorflow:ubuntu16.04-cuda10.0-bazel0.16.1-tensorflow1.12.0
commands:
- "export LD_LIBRARY_PATH=/usr/local/cuda/lib64/stubs:$${LD_LIBRARY_PATH}"
- cd inference/cc/
- cmake . -DPYTHON_EXECUTABLE=python2
- make
environment:
TENSORFLOW_BUILD_DIR: /tensorflow_dist
TENSORFLOW_SOURCE_DIR: /tensorflow