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.

1234567891011121314151617181920212223242526272829303132333435363738
  1. load("//brain/megbrain/lite:flags.bzl","pthread_select")
  2. cc_library(
  3. name = "mgblar",
  4. copts = ["-std=c++14"],
  5. srcs = glob(["src/**/*.cpp"], exclude = ["src/main.cpp"]),
  6. hdrs = glob(["src/**/*.h"]),
  7. includes = ["src"],
  8. features = if_opt([
  9. "no_exceptions",
  10. "no_rtti",
  11. ]),
  12. defines = [
  13. "LITE_BUILD_WITH_MGE=1",
  14. ],
  15. deps = ["//brain/megbrain/lite:lite_static_test"]+
  16. pthread_select(
  17. ["@com_github_gflags_gflags//:gflags_nothreads"],
  18. ["//external:gflags"]
  19. ),
  20. alwayslink = 1,
  21. visibility = ["//visibility:public"],
  22. )
  23. cc_megvii_binary(
  24. name = "load_and_run",
  25. copts = ["-std=c++14"],
  26. srcs = ["src/main.cpp"],
  27. features = if_opt([
  28. "no_exceptions",
  29. "no_rtti",
  30. ]),
  31. internal_deps = [":mgblar"],
  32. visibility = ["//visibility:public"],
  33. )

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