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

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

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