第二阶段测试的时候显存爆了

Open daxiongpro opened this issue 2 years ago • 3 comments

你好，我在第二阶段测试 ./tools/uniad_dist_eval.sh ./projects/configs/stage1_track_map/base_track_map.py ckpts/uniad_base_e2e.pth 8 如果 gpu 为 6 或者 8 ，显存就爆，CUDA out of memory. 但是如果设置 4 以下，则不会出现。

我的GPU 是 A100，80 G 显存。请问是什么原因呀？

Oct 30 '23 07:10 daxiongpro

6或者8不是指的显存大小，指的是gpu的个数

Jul 04 '24 04:07 Feng-Shilong