Pyinstaller 打包pytorch应用（Yolov5为例）

最新推荐文章于 2024-09-09 09:57:15 发布

小熊果汁

最新推荐文章于 2024-09-09 09:57:15 发布

阅读量6.2k

点赞数 4

文章标签： pytorch python 深度学习

本文链接：https://blog.csdn.net/qq_37972108/article/details/121315299

版权

安装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.