Fix cmake build issues with GPU on Linux (#18775)
* Update include file for cmake While attempting to build tensorflow with cmake for gpu the following build error happens: ``` /home/ubuntu/tensorflow/tensorflow/core/platform/default/gpu/cupti_wrapper.h:26:45: fatal error: cuda/extras/CUPTI/include/cupti.h: No such file or directory compilation terminated. CMakeFiles/tf_core_cpu.dir/build.make:3302: recipe for target 'CMakeFiles/tf_core_cpu.dir/home/ubuntu/tensorflow/tensorflow/core/platform/default/gpu/cupti_wrapper.cc.o' failed make[2]: *** [CMakeFiles/tf_core_cpu.dir/home/ubuntu/tensorflow/tensorflow/core/platform/default/gpu/cupti_wrapper.cc.o] Error 1 CMakeFiles/Makefile2:2402: recipe for target 'CMakeFiles/tf_core_cpu.dir/all' failed make[1]: *** [CMakeFiles/tf_core_cpu.dir/all] Error 2 Makefile:127: recipe for target 'all' failed make: *** [all] Error 2 ``` This fix tries to fix the build error above for cmake. Signed-off-by:Yong Tang <yong.tang.github@outlook.com> * Add xla dependency to stream_executor Signed-off-by:
Yong Tang <yong.tang.github@outlook.com> * Remove nccl files temporarily Signed-off-by:
Yong Tang <yong.tang.github@outlook.com> * Add gomp library to libcusolver.so Signed-off-by:
Yong Tang <yong.tang.github@outlook.com> * Fix cmake issue Signed-off-by:
Yong Tang <yong.tang.github@outlook.com>
Loading
Please sign in to comment