How to use PYTORCH_CUDA_ALLOC_CONF=max_split_size_mb: for CUDA out of memory
PYTORCH_CUDA_ALLOC_CONF=max_split_size_mb:
CUDA out of memory