Numpy2Numpy需求极其庞大的显存

[JasonLeon01]

在执行SS的时候报出：
torch.OutOfMemoryError: CUDA out of memory. Tried to allocate 422.62 GiB. GPU 1 has a total capacity of 23.69 GiB of which 4.56 GiB is free. Process 69171 has 12.03 GiB memory in use. Including non-PyTorch memory, this process has 7.08 GiB memory in use. Of the allocated memory 6.81 GiB is allocated by PyTorch, and 14.34 MiB is reserved by PyTorch but unallocated. If reserved but unallocated memory is large try setting PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True to avoid fragmentation. See documentation for Memory Management (https://pytorch.org/docs/stable/notes/cuda.html#environment-variables)

[JasonLeon01]

补充一下，似乎只有MossFormer2_SE_48K正常，其余的我试了FRCRN_SE_16K，也有报错
torch.OutOfMemoryError: CUDA out of memory. Tried to allocate 422.62 GiB. GPU 1 has a total capacity of 23.69 GiB of which 4.56 GiB is free. Process 69171 has 12.03 GiB memory in use. Including non-PyTorch memory, this process has 7.08 GiB memory in use. Of the allocated memory 6.81 GiB is allocated by PyTorch, and 14.34 MiB is reserved by PyTorch but unallocated. If reserved but unallocated memory is large try setting PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True to avoid fragmentation. See documentation for Memory Management (https://pytorch.org/docs/stable/notes/cuda.html#environment-variables)