onnxruntime-gpu用pyinstaller打包报错找不到CUDA解决方案

问题由来:在将深度学习模型转为onnx格式后,由于不需要依赖之前框架环境,仅仅需要由onnxruntime-gpu或onnxruntime即可运行,因此用pyinstaller打包将更加方便。但在实际打包过程中发现,CPU版本的onnxruntime通过pyinstaller打包后生成的exe第三方可以顺利调用,而GPU版本的 onnxruntime-gpu则会出现找不到CUDA报错的情况但实际上电脑已经安装了对应版本的CUDA(如10.1、10.2)。

问题分析:起初博主以为是anaconda环境管理错误等问题,随后通过在python官方下载纯净的python环境后打包,依旧无法找到CUDA。但惊喜的是,博主发现其根本原因是python hook不到onnxruntime-gpu这个包。

解决方案:

  • ​​​​​步骤一首先要注意一点,不能用pyinstaller -F命令,该命令是将全部依赖打包为一个.exe执行文

你可能感兴趣的:(视觉检测图像分割干货,onnxruntime-gpu,打包,pyinstaller,CUDA)