cuda cleanup

avatar
unknown
python
3 years ago
103 B
16
Indexable
# Delete this model to free up GPU memory
del model
import gc
torch.cuda.empty_cache()
gc.collect()
Editor is loading...