Skip to content

Instantly share code, notes, and snippets.

@hewumars

hewumars/error.md

Created January 29, 2021 07:27
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save hewumars/794f7b2c0ed8ff048e7021765f96a8a0 to your computer and use it in GitHub Desktop.
Save hewumars/794f7b2c0ed8ff048e7021765f96a8a0 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
error.md
  1. [ONNXRuntimeError] : 1 : FAIL : Non-zero status code returned while running Div node GPU版时出现，CPU版正常
  2. INVALID_ARGUMENT : Unexpected input data type. 关注输入数据类型
Raw
install.md

从源码编译

  1. CUDA版,Python和C接口

    proxychains ./build.sh --config Release --update --build --build_wheel --use_cuda --cudnn_home /usr/local/cuda/ --cuda_home /usr/local/cuda/ --parallel

  2. TensorRT版

    proxychains ./build.sh --config Release --update --build --build_wheel --cudnn_home /usr/local/cuda/ --cuda_home /usr/local/cuda/ --use_tensorrt --tensorrt_home /home/mars/trt/TensorRT-7.1.3.4/ --parallel

  3. CPU版

    proxychains ./build.sh --config Release --update --build --build_wheel --parallel

wheel文件在build/Linux/Release/dist/，安装Python whl时，注意先uninstall，再install新的whl。

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment