You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

CMakeLists.txt 1.4 kB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647
  1. file (GLOB_RECURSE SOURCES ./*.cpp)
  2. add_executable(lite_examples ${SOURCES})
  3. if(LITE_BUILD_WITH_RKNPU)
  4. #rknn sdk1.0.0 depend on libc++_shared, use gold to remove NEEDED so symbol check
  5. target_link_options(lite_examples PRIVATE "-fuse-ld=gold")
  6. endif()
  7. target_link_libraries(lite_examples lite_static)
  8. if(LITE_BUILD_WITH_MGE AND MGE_WITH_ROCM)
  9. # FIXME: hip obj can not find cpp obj only through lite_static
  10. target_link_libraries(lite_examples megdnn)
  11. endif()
  12. if(UNIX)
  13. if(APPLE OR ANDROID)
  14. target_link_libraries(lite_examples dl)
  15. else()
  16. target_link_libraries(lite_examples dl rt)
  17. endif()
  18. endif()
  19. install (TARGETS lite_examples
  20. EXPORT ${LITE_EXPORT_TARGETS}
  21. RUNTIME DESTINATION lite/bin)
  22. # add lite_examples_depends_shared for CI check symbol export valid
  23. add_executable(lite_examples_depends_shared ${SOURCES})
  24. if(LITE_BUILD_WITH_RKNPU)
  25. #rknn sdk1.0.0 depend on libc++_shared, use gold to remove NEEDED so symbol check
  26. target_link_options(lite_examples_depends_shared PRIVATE "-fuse-ld=gold")
  27. endif()
  28. target_link_libraries(lite_examples_depends_shared lite_shared)
  29. if(UNIX)
  30. if(APPLE OR ANDROID)
  31. target_link_libraries(lite_examples_depends_shared dl)
  32. else()
  33. target_link_libraries(lite_examples_depends_shared dl rt)
  34. endif()
  35. endif()
  36. install (TARGETS lite_examples_depends_shared
  37. EXPORT ${LITE_EXPORT_TARGETS}
  38. RUNTIME DESTINATION lite/bin)

MegEngine 安装包中集成了使用 GPU 运行代码所需的 CUDA 环境,不用区分 CPU 和 GPU 版。 如果想要运行 GPU 程序,请确保机器本身配有 GPU 硬件设备并安装好驱动。 如果你想体验在云端 GPU 算力平台进行深度学习开发的感觉,欢迎访问 MegStudio 平台