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.

MegEngineConfig.cmake.in 343 B

1234567891011121314
  1. @PACKAGE_INIT@
  2. find_package(Threads)
  3. find_package(Flatbuffers REQUIRED)
  4. if (@MGE_WITH_MKLDNN@)
  5. find_package(dnnl REQUIRED)
  6. endif()
  7. list (APPEND CMAKE_MODULE_PATH ${CMAKE_CURRENT_LIST_DIR})
  8. if (NOT TARGET megengine)
  9. include ("${CMAKE_CURRENT_LIST_DIR}/@MGE_EXPORT_TARGETS@.cmake")
  10. check_required_components("megengine")
  11. endif ()

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