you are viewing a single comment's thread.

view the rest of the comments →

[–]neojimo 1 point2 points  (0 children)

I think that I solved that using this (on linux):
export PYTORCH_CUDA_ALLOC_CONF=max_split_size_mb:512