yolov7 の export.py を変換すると、メモリ エラーが報告されます。予約メモリが >> 割り当てメモリの場合は、max_split_size_mb を a に設定してみてください。

内容:エクスポート失敗:CUDA のメモリ不足。20.00 MiB を割り当てようとしました (GPU 0、合計容量 4.00 GiB、すでに割り当てられている 2.45 GiB、空き 0 バイト、PyTorch によって合計 2.54 GiB が予約されています) 予約メモリ >> 割り当てメモリの場合は、断片化を避けるために max_split_size_mb を設定してみてください。メモリ管理と PYTORCH_CUDA_ALLOC_CONF のドキュメントを参照してください。

解決策:
追加

import os
os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "max_split_size_mb:128"

そうでない場合は、この文を include-nms に追加してください

default=['torchscript'],

ここに画像の説明を挿入

おすすめ

転載: blog.csdn.net/weixin_44021553/article/details/128223155