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.

utils.sh 1.7 kB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071
  1. #!/usr/bin/env bash
  2. set -e
  3. MAKEFILE_TYPE="Unix"
  4. OS=$(uname -s)
  5. if [[ $OS =~ "NT" ]]; then
  6. echo "BUILD in NT ..."
  7. MAKEFILE_TYPE="Unix"
  8. fi
  9. function build_flatc() {
  10. BUILD_DIR=$1/build_dir/host_flatc/build
  11. INSTALL_DIR=$BUILD_DIR/../install
  12. REMOVE_OLD_BUILD=$2
  13. if [ $REMOVE_OLD_BUILD = "true" ]; then
  14. if [ -e $BUILD_DIR ];then
  15. echo "clean old dir: $BUILD_DIR"
  16. rm -rf $BUILD_DIR
  17. fi
  18. if [ -e $INSTALL_DIR ];then
  19. echo "clean old dir: $INSTALL_DIR"
  20. rm -rf $INSTALL_DIR
  21. fi
  22. else
  23. echo "strip remove old build"
  24. fi
  25. if [ ! -e $BUILD_DIR ];then
  26. echo "create build: $BUILD_DIR"
  27. mkdir -p $BUILD_DIR
  28. fi
  29. if [ ! -e $INSTALL_DIR ];then
  30. echo "create install $INSTALL_DIR"
  31. mkdir -p $INSTALL_DIR
  32. fi
  33. cd $BUILD_DIR
  34. cmake -G "$MAKEFILE_TYPE Makefiles" \
  35. -DCMAKE_BUILD_TYPE=Release \
  36. -DCMAKE_INSTALL_PREFIX=$INSTALL_DIR \
  37. -DFLATBUFFERS_BUILD_TESTS=OFF \
  38. -DFLATBUFFERS_BUILD_FLATHASH=OFF \
  39. -DFLATBUFFERS_BUILD_FLATLIB=OFF \
  40. -DFLATBUFFERS_LIBCXX_WITH_CLANG=OFF \
  41. $SRC_DIR/third_party/flatbuffers
  42. make -j$(nproc)
  43. make install/strip
  44. }
  45. function try_remove_old_build() {
  46. REMOVE_OLD_BUILD=$1
  47. echo $REMOVE_OLD_BUILD
  48. BUILD_DIR=$2
  49. INSTALL_DIR=$3
  50. if [ $REMOVE_OLD_BUILD = "true" ]; then
  51. if [ -e $BUILD_DIR ];then
  52. echo "clean old dir: $BUILD_DIR"
  53. rm -rf $BUILD_DIR
  54. fi
  55. if [ -e $INSTALL_DIR ];then
  56. echo "clean old dir: $INSTALL_DIR"
  57. rm -rf $INSTALL_DIR
  58. fi
  59. else
  60. echo "strip remove old build"
  61. fi
  62. }

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