GPT训练，3090ti训练都不够为什么?

RVC-Boss / GPT-SoVITS

1 min voice data can also be used to train a good TTS model! (few shot voice cloning)

MIT License

36.34k stars 4.15k forks source link

Closed yeloveyou closed 9 months ago

yeloveyou commented 10 months ago

gpt训练卡顿，并且进度条不显示，占用率100%，这是为什么呢？（我的训练文件就两个，合起来1分钟左右）

BlacknoSheep commented 10 months ago

打开cmd，输入 nvidia-smi 看看具体是哪些进程在占显存，我 batch_size=16 时SoVITS 约占 13.5G 显存，GPT 约占 14.5G 显存。显存不够就等SoVITS训练完了再训练GPT。

bfloat16 commented 10 months ago

切片太长，共享显存占用严重

RVC-Boss commented 9 months ago

楼上正解。需要关闭共享显存。https://www.bilibili.com/read/cv27436189 参考这个，找到你调用的python.exe关闭一下。