search

Sharpiless / Yolov5-Deepsort

最新版本yolov5+deepsort目标检测和追踪，能够显示目标类别，支持5.0版本可训练自己数据集
GNU General Public License v3.0
959 stars 159 forks source link

显存越来越高 #8

Open nobody-cheng opened 2 years ago

nobody-cheng commented 2 years ago

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

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

oddmanru commented 1 year ago

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

TryHard-LL commented 1 year ago

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

oddmanru commented 1 year ago

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

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

TryHard-LL commented 1 year ago

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