背景:这阵子用cpu训练不到6000张的数据集,训练50个epoch,花了27个小时,就是连上两个工作日。自己有个好几年前的4G的1050ti。对比一下。
batch_size | 显存占用 | 内存占用 | 训练时间/epoch | |
cpu | 8 | 0 MB | --GB | 32 分钟 左右 |
1050 TI gpu | 4 | 1386MB | 7 GB 左右 | 3.6 分钟 左右 |
8 | 2014MB | 8 GB 左右 | 2.8 分钟 左右 |
1050 ti gpu 下 , batch_size = 4。
1050 TI gpu 下 , batch_size = 8。
据说,10张1050ti显卡也比不上一张3090ti显卡。那现在一块好显卡,它可真快。
奈何,我的笔记本内存是8G的,显卡它没充分使用上。