onnxruntime-gpu用pyinstaller打包报错找不到CUDA

onnxruntime

microsoft/onnxruntime: 是一个用于运行各种机器学习模型的开源库。适合对机器学习和深度学习有兴趣的人，特别是在开发和部署机器学习模型时需要处理各种不同框架和算子的人。特点是支持多种机器学习框架和算子，包括 TensorFlow、PyTorch、Caffe 等，具有高性能和广泛的兼容性。

项目地址：https://gitcode.com/gh_mirrors/on/onnxruntime

免费下载资源

wdmzlyq

1424人浏览 · 2023-06-19 16:06:58

wdmzlyq · 2023-06-19 16:06:58 发布

pyinstaller 打包会报onnxruntime-gpu时缺少onnxruntime-cuda的动态库。打包时加上动态库就可以了。

使用--add-binary。如下图所示，首先第一入参是依赖库的位置，第二个入参是打包exe指定依赖库的位置

pyinstaller  --add-binary "D:/soft/Anaconda3/envs/pyinstaller/Lib/site-packages/onnxruntime/capi/onnxruntime_providers_cuda.dll;./onnxruntime/capi" --add-binary "D:/soft/Anaconda3/envs/pyinstaller/Lib/site-packages/onnxruntime/capi/onnxruntime_providers_shared.dll;./onnxruntime/capi"

GitHub 加速计划 / on / onnxruntime

下载

最近提交(Master分支：5 个月前 )

97c2bbe3 ### Description  Fix shape infer of onnx GroupNorm. ### Motivation and Context  Unable to run shape inference for onnx `GroupNorm`. [model.onnx](https://raw.githubusercontent.com/onnx/onnx/refs/heads/main/onnx/backend/test/data/node/test_group_normalization_example/model.onnx) > python D:\source\cognition\onnxruntime\onnxruntime\python\tools\symbolic_shape_infer.py --input model.onnx Traceback (most recent call last): File "D:\source\cognition\onnxruntime\onnxruntime\python\tools\symbolic_shape_infer.py", line 2999, in <module> out_mp = SymbolicShapeInference.infer_shapes( File "D:\source\cognition\onnxruntime\onnxruntime\python\tools\symbolic_shape_infer.py", line 2935, in infer_shapes raise Exception("Incomplete symbolic shape inference") 1 天前

1fc9c482 ### Description Enable coremltools for Linux build. In order to do this, I did: 1. Add uuid-devel to the Linux images and regenerate them. 2. Patch the coremltools code a little bit to add some missing header files. ### Motivation and Context To make the code simpler. Later on I will create another PR to remove the COREML_ENABLE_MLPROGRAM C/C++ macro. Also, after this PR I will bring more changes to onnxruntime_provider_coreml.cmake to make it work with vcpkg. 2 天前