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

OpenDriveLab / UniAD

[CVPR'23 Best Paper Award] Planning-oriented Autonomous Driving

Apache License 2.0

3.11k stars 335 forks source link

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

Open daxiongpro opened 8 months ago

daxiongpro commented 8 months ago

你好，我在第二阶段测试 ./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 显存。请问是什么原因呀？

Feng-Shilong commented 1 week ago

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