背景:这阵子用cpu训练不到6000张的数据集,训练50个epoch,花了27个小时,就是连上两个工作日。自己有个好几年前的4G的1050ti。对比一下。

batch_size显存占用内存占用训练时间/epoch
cpu80 MB--GB32 分钟 左右

1050 TI gpu

41386MB7 GB 左右3.6 分钟 左右
82014MB8 GB 左右2.8 分钟 左右

1050 ti gpu 下 , batch_size = 4。

1050 TI gpu 下 , batch_size = 8。 

 

据说,10张1050ti显卡也比不上一张3090ti显卡。那现在一块好显卡,它可真快。

奈何,我的笔记本内存是8G的,显卡它没充分使用上。

 

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐