训练LORA相关
stablediffusion吧
全部回复
仅看楼主
level 6
最终Anubis 楼主
爆显存了,提示如下
CUDA out of memory. Tried to allocate 10.00 MiB. GPU 0 has a total capacity of 16.00 GiB of which 8.87 GiB is free. Of the allocated memory 5.88 GiB is allocated by PyTorch, and 72.50 MiB is reserved by PyTorch but unallocated. If reserved but unallocated memory is large try setting PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True to avoid fragmentation. See documentation for Memory Management
显卡4060ti16g
其中提到了,16G显存,其中8.87G是空闲的?
需要去哪里设置开启全部显存呢?
2024年04月19日 01点04分 1
level 13
占用多少只跟你训练分辨率有关,1024*1024训练也就占10来个G,16还能爆肯定毛优化都没做
2024年04月19日 02点04分 2
训练用的分辨率全是512*512,秋叶的LORA训练器。.主要是我看报错提示,按它意思我有8.87G是空闲的,是什么没设置好导致它只用了一半的显存吗?
2024年04月19日 05点04分
@最终Anubis 啥都不会改的你最好去换个炉子,你看看行者丹炉,还不用百度云下载有自动安装程序
2024年04月19日 06点04分
1