RuntimeError: CUDA out of memory. How can I set max_split_size_mb?
The max_split_size_mb configuration value can be set as an environment variable. The exact syntax is documented, but in short: The behavior of caching allocator can be controlled via environment variable PYTORCH_CUDA_ALLOC_CONF. The format is PYTORCH_CUDA_ALLOC_CONF=<option>:<value>,<option2>:<value2>… Available options: … max_split_size_mb prevents the allocator from splitting blocks larger than this size (in MB). This can help prevent … Read more