windows pyinstaller打包pytorch程序到exe文件

主要内容

利用pyinstaller将pytorch代码打包成exe其实含有比较多大的坑，这篇文章主要记录最近在项目中打包exe遇到的一些问题，由于问题主要是opencv和pytorch带来的，解决的办法和minist的pytorch程序差不多，这个minist程序我已经测试过，能够打包成功，并在没有python环境的电脑上成功运行：https://github.com/1240117300/MINIST。

出坑技巧一：安装匹配的包版本

这是虚拟环境中的安装包及对应的版本号，这张表就能差不多解决问题了，很多奇奇怪怪的问题，查了一堆资料之后，发现换一个包的版本就解决了。

出坑技巧二：从干净的虚拟环境中，只安装需要的包，能够有效减小打包文件的大小，并提高打包速度

这些安装目录在百度磁盘有，下载之后能够快速的部署深度学习环境

出坑技巧三：编写好spec文件

# -*- mode: python ; coding: utf-8 -*-import sys
import os.path as osp
sys.setrecursionlimit(5000)block_cipher = Nonea = Analysis(['airway_seg.py'],pathex=['E:\\MikeyNa\\pycode\\airtree_seg_xajd\\airway_deploy','E:\\MikeyNa\\pycode\\airtree_seg_xajd\\airway_deploy\\preprocessing','D:\\deeplearning\\Anaconda3\\envs\\airway_deploy\\Lib\\site-packages'],binaries=[],datas=[('baseline_fr_ad.ckpt','baseline_fr_ad.ckpt')],hiddenimports=['pydicom','models','preprocessing','full_prep','baseline_fr','data','option','prepare','split_combine_mj','trainval_classifier','utils'],hookspath=[],hooksconfig={},runtime_hooks=[],excludes=[],win_no_prefer_redirects=False,win_private_assemblies=False,cipher=block_cipher,noarchive=False)
pyz = PYZ(a.pure, a.zipped_data,cipher=block_cipher)exe = EXE(pyz,a.scripts, [],exclude_binaries=True,name='airway_seg',debug=False,bootloader_ignore_signals=False,strip=False,upx=True,console=True,disable_windowed_traceback=False,target_arch=None,codesign_identity=None,entitlements_file=None )
coll = COLLECT(exe,a.binaries,a.zipfiles,a.datas, strip=False,upx=True,upx_exclude=[],name='airway_seg')

出坑技巧四：从坑里爬出来，下面记录一些我遇到遇到的具体坑和解决办法

Multiprocessing error

windows下，当代码引入multiprocessing模块并使用pyinstaller打包时，会遇到exe死锁状态

原因：windows下multiprocessing模块设置有问题，在linux下没有这个问题，解决办法

from multiprocessing import freeze_supportif __name__ == '__main__':start_time = time.time()freeze_support()main()end_time = time.time()print('end airway segmentation, time %d seconds' % (end_time - start_time))

参考链接

使用pyinstaller打包多文件和目录的Python项目 - 是勇哥呀 - 博客园

总结使用pyinstaller打包pyqt5以及pytorch的各种坑_Nie_Hen的博客-CSDN博客_pyinstaller torch

嘗試使用Pyinstaller打包yolov5中detect.py文件(含pytorch框架)所遇到的問題

https://medium.com/@exexzxzpo/%E5%98%97%E8%A9%A6%E4%BD%BF%E7%94%A8pyinstaller%E6%89%93%E5%8C%85yolov5%E4%B8%ADdetect-py%E6%96%87%E4%BB%B6-%E5%90%ABpytorch%E6%A1%86%E6%9E%B6-%E6%89%80%E9%81%87%E5%88%B0%E7%9A%84%E5%95%8F%E9%A1%8C-543f5481b304