There was an error while loading. Please reload this page.
1 parent 63bc266 commit b67f3c8Copy full SHA for b67f3c8
CMakeLists.txt
@@ -41,8 +41,8 @@ ExternalProject_Add(${PROJECT_NAME}-ext
41
TF_CUDA_COMPUTE_CAPABILITIES=3.5,5.2
42
./configure
43
44
- BUILD_COMMAND bazel build -c opt -j 200 ${TF_BUILD_CUDA} //tensorflow:libtensorflow.so
45
- //tensorflow/tools/pip_package:build_pip_package
+ BUILD_COMMAND bazel build -c opt -j 200 --config=opt ${TF_BUILD_CUDA} //tensorflow:libtensorflow.so
+ //tensorflow/tools/pip_package:build_pip_package
46
INSTALL_COMMAND cp <SOURCE_DIR>/bazel-bin/tensorflow/libtensorflow.so ${CMAKE_SOURCE_DIR}/lib
47
COMMAND <SOURCE_DIR>/bazel-bin/tensorflow/tools/pip_package/build_pip_package ${CMAKE_SOURCE_DIR}/lib
48
COMMAND /bin/bash -c "wheel unpack ${CMAKE_SOURCE_DIR}/lib/tensorflow-*.whl"
0 commit comments