Yolov5-Deepsort icon indicating copy to clipboard operation
Yolov5-Deepsort copied to clipboard

显存越来越高

Open nobody-cheng opened this issue 3 years ago • 4 comments

使用的rtx3060 12g显存 centos7 cuda:11.1,驱动:470.63.01 torch:1.7.1+cu110

接RTSP视频流,开始占用显存大概1.3G,过一段时间后显存占用越来越高,直到显存溢出,屏蔽deepsort部分,只跑检测不会出现这情况,不知道deepsort哪一部分导致显存占用升高

nobody-cheng avatar Feb 16 '22 09:02 nobody-cheng

显卡占用高低与性能之间没有必然联系。文章中这段程序,DeepSort所起到的作用很有限,不是占用资源的主要地方。

oddmanru avatar Nov 20 '22 07:11 oddmanru

可能是这其中的某个变量一直在累积,致使随着迭代,不断增加~

TryHard-LL avatar Nov 24 '22 06:11 TryHard-LL

可能是这其中的某个变量一直在累积,致使随着迭代,不断增加~

他遇到的情况是显存占用高,而不是内存占用高。变量的积累影响的是内存而不会是现存。在我看来,Deep learning 导致显存溢出先要检查batch size 设置是否正确。不过DeepSORT在本文中的作用非常有限。

oddmanru avatar Nov 24 '22 07:11 oddmanru

不一定,因为你有的变量是放在gpu上的,在训练过程中,一直会在这上面进行累积

TryHard-LL avatar Nov 24 '22 07:11 TryHard-LL