Pyinstaller 打包pytorch应用（Yolov5为例）

安装pytorch

pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html

**注：**打包cudatoolkit会出错（报错如图1所示，如果有解决办法请打在评论区，谢谢！），因此使用pip安装pytorch并手动配置CUDA.

打包应用

# 安装pyinstaller
pip install pyinstaller

# 使用pyinstaller打包应用（run.py为程序入口文件）
pyinstaller --noconfirm --onefile --console --clean  run.py

遇到的坑

以Yolov5为例

• Linux下需要将pin_memery = False。
• 切记一定要关闭360（如果打开360可能会报如下错误）！！！
  
• torch_uilt.py: select_device函数中将 “git_describe() or date_modified()” 改为 “date_modified()”。
• pathlib 尽量使用os.path替代
• 加入multiprocessing.freeze_support()。
  
• opencv-python==4.5.3.56 （版本过高打包完运行时会报错）。
• 打包cudatoolkit会出错（报错如图1所示，如果有解决办法请打在评论区，谢谢！），因此使用pip安装pytorch并手动配置CUDA.