Keep getting CUDA OOM error with Pytorch failing to allocate all free memory

[logy]

@sbelharbi

export PYTORCH_CUDA_ALLOC_CONF=max_split_size_mb:128

excuse me did you solve it ?