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.

FetchMegBrainVersion.cmake 1.1 kB

123456789101112131415161718192021222324252627282930313233
  1. # Parses the version set in src/core/include/megbrain/version.h
  2. # Exports the following variables:
  3. # MGB_VER_MAJOR: Major version
  4. # MGB_VER_MINOR: Minor version
  5. # MGB_VER_PATCH: Patch version
  6. # MGB_IS_DEV: Is development version
  7. # MGB_VER_STRING: Version string
  8. option(MGB_FORCE_DEV_VERSION "Force -dev tag in version stamp" OFF)
  9. file (READ "${CMAKE_SOURCE_DIR}/src/core/include/megbrain/version.h" content)
  10. string (REGEX MATCH "MGB_MAJOR +([0-9]+)" _ ${content})
  11. set (MGB_VER_MAJOR ${CMAKE_MATCH_1})
  12. string (REGEX MATCH "MGB_MINOR +([0-9]+)" _ ${content})
  13. set (MGB_VER_MINOR ${CMAKE_MATCH_1})
  14. string (REGEX MATCH "MGB_PATCH *([0-9]+)" _ ${content})
  15. set (MGB_VER_PATCH ${CMAKE_MATCH_1})
  16. if (MGB_FORCE_DEV_VERSION)
  17. set (MGB_IS_DEV 1)
  18. else()
  19. string (REGEX MATCH "MGB_IS_DEV +([01])" _ ${content})
  20. set (MGB_IS_DEV ${CMAKE_MATCH_1})
  21. endif()
  22. set (MGB_VER_STRING "${MGB_VER_MAJOR}.${MGB_VER_MINOR}.${MGB_VER_PATCH}")
  23. if (MGB_IS_DEV)
  24. set (MGB_VER_STRING "${MGB_VER_STRING}-dev")
  25. endif()
  26. message(STATUS "Building MegBrain ${MGB_VER_STRING}")

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