bubbliiiing / yolov7-pytorch

这是一个yolov7的库,可以用于训练自己的数据集。
GNU General Public License v3.0
861 stars 150 forks source link

导,为什么在相同环境下,yolov5可以跑,但是yolov7却出现了这样的问题? #59

Open 1091448676 opened 1 year ago

1091448676 commented 1 year ago

image

bubbliiiing commented 1 year ago

嗯?看起来都是cpu?

ZJU-heyike commented 1 year ago

我也出现了这个问题,我用的是线上服务器的gpu,跑yolov5和yolox都是好的,请问你解决了吗?

bubbliiiing commented 1 year ago

你是什么版本

ZJU-heyike commented 1 year ago

我的torch的版本是1.13,对应的cuda是11.7,显卡是4张3090,也是出现这个问题(我看网上有说是多显卡的原因,还有说降低torch的版本,不知道到底是哪个原因呀,急需我b导帮助点拨解决一下)

bubbliiiing commented 1 year ago

大概是torch版本的问题,你把那个fg mask inboxes的device设置成和from which tensor一样就可以

bubbliiiing commented 1 year ago

主要我受伤没有1.13的机器

ZJU-heyike commented 1 year ago

解决啦,感谢b导~

bubbliiiing commented 1 year ago

我自己也改一下,蛮多人问的