在DCU2的环境上进行训练报错提示显存不够
torch.cuda.OutOfMemoryError: HIP out of memory. Tried to allocate 26.00 MiB. GPU 0 has a total capacty of 15.98 GiB of which 0 bytes is free. Of the allocated memory 15.70 GiB is allocated by PyTorch, and 52.62 MiB is reserved by PyTorch but unallocated. If reserved but unallocated memory is large try setting max_split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_HIP_ALLOC_CONF 请问您是在哪个环境把这个模型跑通的，遇到这个问题除了换环境还有其他解决办法吗