Usage of max_split_size_mb

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

1 Like