CUDA out of memory #28
Description
adata = model.fit(lr=lr, epochs=epochs,)
The CUDA memory is insufficient. How can we reduce its usage?
OutOfMemoryError: CUDA out of memory. Tried to allocate 222.41 GiB (GPU 0; 23.56 GiB total capacity; 583.94 MiB already allocated; 22.66 GiB free; 612.00 MiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation.