Al convertir export.py de yolov7, se informa el error de memoria. Si la memoria reservada es ＞＞ memoria asignada, intente establecer max_split_size_mb en un

报错内容：Error de exportación: CUDA sin memoria. Intenté asignar 20.00 MiB (GPU 0; 4.00 GiB de capacidad total; 2.45 GiB ya asignados; 0 bytes libres; 2.54 GiB reservados en total por PyTorch) Si la memoria reservada es >> memoria asignada, intente configurar max_split_size_mb para evitar la fragmentación. Consulte la documentación para la gestión de memoria y PYTORCH_CUDA_ALLOC_CONF

Solución:
agregar

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

Si no, agregue esta oración a include-nms

default=['torchscript'],