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.4 kB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859
  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. if [ -e $BUILD_DIR ];then
  13. echo "clean old dir: $BUILD_DIR"
  14. rm -rf $BUILD_DIR
  15. fi
  16. if [ -e $INSTALL_DIR ];then
  17. echo "clean old dir: $INSTALL_DIR"
  18. rm -rf $INSTALL_DIR
  19. fi
  20. echo "create build dir"
  21. mkdir -p $BUILD_DIR
  22. mkdir -p $INSTALL_DIR
  23. cd $BUILD_DIR
  24. cmake -G "$MAKEFILE_TYPE Makefiles" \
  25. -DCMAKE_BUILD_TYPE=Release \
  26. -DCMAKE_INSTALL_PREFIX=$INSTALL_DIR \
  27. -DFLATBUFFERS_BUILD_TESTS=OFF \
  28. -DFLATBUFFERS_BUILD_FLATHASH=OFF \
  29. -DFLATBUFFERS_BUILD_FLATLIB=OFF \
  30. -DFLATBUFFERS_LIBCXX_WITH_CLANG=OFF \
  31. $SRC_DIR/third_party/flatbuffers
  32. make -j$(nproc)
  33. make install/strip
  34. }
  35. function try_remove_old_build() {
  36. REMOVE_OLD_BUILD=$1
  37. echo $REMOVE_OLD_BUILD
  38. BUILD_DIR=$2
  39. INSTALL_DIR=$3
  40. if [ $REMOVE_OLD_BUILD = "true" ]; then
  41. if [ -e $BUILD_DIR ];then
  42. echo "clean old dir: $BUILD_DIR"
  43. rm -rf $BUILD_DIR
  44. fi
  45. if [ -e $INSTALL_DIR ];then
  46. echo "clean old dir: $INSTALL_DIR"
  47. rm -rf $INSTALL_DIR
  48. fi
  49. else
  50. echo "strip remove old build"
  51. fi
  52. }

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