最近接触了在python用最常用的Pyinstaller打包exe文件,需要将pytorch模型和pyqt写的界面打包。用到的外部库主要是onnxruntime(cpu)、pyqt5、opencv和numpy。(一开始在pytorch环境下直接打包,最后exe文件1.33G大,离谱!!所以改成用onnxruntime调用onnx格式的模型)
打包exe
因为预感打包会出现各种问题,提前搜了一些资料:
步骤:
1、新建虚拟环境,只安装需要的库;
2、直接Pyinstaller -F main.py 打包,没有使用spec文件;
3、打包出来的exe文件83M左右;
4、修改exe文件的位置,解决了一些路径问题,程序可以运行,Qt界面也可以正常交互,但是但是,会提示找不到onnxruntime_providers_shared.dll 文件。
问题解决
本来想想找不到文件就算了,反正我电脑上可以成功运行。但是程序移植到新电脑上时,会报没有VC环境之类的错误,直接就给我闪退了。
为了解决这个烦人的问题,就参考上面的博客1,修改spec文件来打包,结果折腾了一晚上还是没解决。于是痛定思痛,第二天早上查看Pyinstaller官方介绍,发现可以把这个dll文件加进去!!
按这上面的指示,修改main.spec文件,主要改动如下:
a = Analysis(['main.py'],
pathex=[],
binaries=[],
datas=[('D:\\anaconda3\\envs\\xffpackage\\Lib\\site-packages\\onnxruntime\\capi\\onnxruntime_providers_shared.dll', 'onnxruntime\\capi')],
hiddenimports=[],
hookspath=[],
hooksconfig={},
runtime_hooks=[],
excludes=[],
win_no_prefer_redirects=False,
win_private_assemblies=False,
cipher=block_cipher,
noarchive=False)
最后再 Pyinstaller -F main.spec ,再运行打包后的文件,就不会报缺少dll了,舒服了!!
其他解决办法
在看官方介绍的时候发现这样一段话:
大概是说 绑定到单个文件时(exe文件),运行该文件时会生成一个临时的MEIxxxxxx 的文件夹,也就是上面报错的那个MEI112602文件夹,本来直接找到报错的那个目录,把dll文件加进去应该就完事了,但是这个文件夹程序关闭就会删掉,是临时的!
所以应该可以选择 绑定到一个文件夹的方式(Bundling to One Folder)打包程序,最后把dll文件加到报错的那个目录就能解决问题。
不过以上只是本人的猜想,没有实际验证,仅供参考。
移植出现的新问题及解决方案
本来以为解决了dll的问题就万事大吉了,结果移植到新电脑上有出现了问题,裂开。。。
总共出现了2个问题:
- 路径问题,导入自己的写的py脚本出错
- 还是缺少VC环境,报错:Microsoft Visual C++ Redistributable for Visual Studio 2019 not installed on the machin
针对问题1,我之前的写法是用os.getcwd()获取当前工作路径,并把它和相应的包文件路径都加到搜索路径里,在自己电脑上ok,移植后os.getcwd()就不是.exe所在路径了,就不ok了,不知道为啥。代码如下:
try:
from PyQt5 import QtCore, QtGui, QtWidgets
import os
import sys
root = os.getcwd()
print('root: ', root)
sys.path.append(os.path.join(root, 'camera'))
sys.path.append(os.path.join(root, 'detect'))
print('Path: ', sys.path)
os.chdir(root)
print('cwd: ', os.getcwd())
from third_version import Ui_MainWindow
except Exception as e:
import traceback
print('主程序导入出错: ', e)
exc_type, exc_value, exc_obj = sys.exc_info()
traceback.print_tb(exc_obj, file=sys.stdout)
input('按任意键退出: ')
参考博客:python-pyinstaller、打包后获取路径,代码改为下面所示,就OK了
try:
from PyQt5 import QtCore, QtGui, QtWidgets
import os
import sys
root = os.path.dirname(os.path.realpath(sys.argv[0]))
print('root: ', root)
sys.path.append(os.path.join(root, 'camera'))
sys.path.append(os.path.join(root, 'detect'))
print('Path: ', sys.path)
os.chdir(root)
print('cwd: ', os.getcwd())
from third_version import Ui_MainWindow
except Exception as e:
import traceback
print('主程序导入出错: ', e)
exc_type, exc_value, exc_obj = sys.exc_info()
traceback.print_tb(exc_obj, file=sys.stdout)
input('按任意键退出: ')
针对问题2,参考博客,下载相应组件安装就ok了!