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

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

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