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 995 B

123456789101112131415161718192021222324
  1. include_directories(src)
  2. file(GLOB_RECURSE SOURCES src/*.cpp main.cpp)
  3. add_executable(load_and_run ${SOURCES})
  4. target_link_libraries(load_and_run megbrain megdnn ${MGE_CUDA_LIBS})
  5. # load_and_run_depends_shared always for CI check, please do not delete
  6. if(BUILD_SHARED_LIBS)
  7. add_executable(load_and_run_depends_shared ${SOURCES})
  8. target_link_libraries(load_and_run_depends_shared megengine)
  9. if(WIN32 OR MSVC)
  10. target_compile_definitions(load_and_run_depends_shared PRIVATE MGE_DLL_IMPORT_DATA)
  11. endif()
  12. endif()
  13. install(TARGETS load_and_run EXPORT ${MGE_EXPORT_TARGETS} RUNTIME DESTINATION ${CMAKE_INSTALL_BINDIR})
  14. if(BUILD_SHARED_LIBS)
  15. install(TARGETS load_and_run_depends_shared EXPORT ${MGE_EXPORT_TARGETS} RUNTIME DESTINATION ${CMAKE_INSTALL_BINDIR})
  16. endif()
  17. if(MGE_WITH_TEST)
  18. add_executable(json_loader_test test/json_loader_test.cpp src/json_loader.h src/json_loader.cpp)
  19. target_link_libraries(json_loader_test megbrain megdnn ${MGE_CUDA_LIBS})
  20. endif()

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