测试环境:
windows10
cuda11.1.1
cudnn8.2.0
tensorrt8.2.3.0
VS2019
onnxruntime==1.12.1
cmake==3.24.3
编译流程:
git clone --recursive https://github.com/Microsoft/onnxruntime
打开:x64 Native Tools Command Prompt for VS 2019
cd /d [youronnxrumtimepath]
.\build.bat --build_shared_lib --parallel --use_cuda --cuda_version 11.1 --cuda_home "C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1" --cudnn_home "C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1" --build_csharp --use_tensorrt --tensorrt_home "D:\lufiles\TensorRT-8.2.3.0" --cmake_generator "Visual Studio 16 2019" --skip_tests --config Release
特别注意:
(1)不要用VS2022编译,要用VS2019
(2)cmake版本不能过高,cmake==3.27.7编译会出现MSB3271 ml64.exe错误,需要cmake==3.24.3或者低一点也行。