纵使你换了3090ti,24g内存,batch_size=16,但是yolov5训练依旧报错。
RuntimeError: CUDA out of memory。
从16调到8,还是报错。
再调到4,可以了,但是只用了7个G的显存。
其实占用率不高,也不知道为啥。。

10.7,国庆最后一天,在学土堆的pytorch
10.8,努力搬砖一天。
10.9,测试显卡,发出咔咔声。
10.10,寄回影驰,晚上旺顺阁吃鱼吃饼。
10.11 ,日常工作干饭。
10.12,显卡换新,寄回。
10.13,晚上踢足球,加班,测试显卡。
10.14-16,配置VITIS AI环境,人麻了已经,还没配好。
10.17-18,学土堆的yolov5.


本文由 sdh 创作,采用 知识共享署名 3.0,可自由转载、引用,但需署名作者且注明文章出处。

只有地板了

  1. nash2zhao
    nash2zhao

    站在高高的土堆旁边

添加新评论