ubuntu下Mxnet的c++编译

最新推荐文章于 2021-05-08 12:22:08 发布

转载最新推荐文章于 2021-05-08 12:22:08 发布 · 1.2k 阅读

收录于

Ubuntu安装Mxnet请参见安装Mxnet 。

将CMakeLists.txt文件修改

cmake_minimum_required(VERSION 2.8)
project(mxnet_cpp_test)
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -g -std=c++11 -W")

include_directories(
    /home/user/mxnet/3rdparty/dmlc-core/include
    /home/user/mxnet/3rdparty/nnvm/include
    /home/user/mxnet/include
    /home/user/mxnet/cpp-package/include
)

aux_source_directory(./src DIR_SRCS)

link_directories(/home/lancer/mxnet/lib)

add_executable(mxnet_cpp_test ${DIR_SRCS})
target_link_libraries(mxnet_cpp_test mxnet)

修改完成就可以运行了。