0%

相关环境

名称 版本
Ubuntu 22.04.1
Python 3.9.7
PyTorch 2.0.1+cu117

解决方法

问题在于utils.generalnon_max_suppression函数中

1
x[:, 5:] *= x[:, 4:5]  # conf = obj_conf * cls_conf

这一行的相乘操作创建的中间变量未能在函数执行完毕后自动释放,导致显存占用无限增长。

将其更改为:

1
2
temp = x[:, 4:5].clone()
x *= temp

即可及时释放显存,解决显存占用无限增长问题。

在中国石油的官网上找到了这个石油知识的栏目,挺适合石油人来回忆过去、评说现在、展望未来。希望这个栏目能一直做下去。正好最近在玩flask,顺便把这个栏目的内容抓了下来做了个简易的网站。也希望我的网站能一直开下去。

石油人无不怀念石油部,我自然也不例外。直到经历了疫情时期,我才发现这原来也许是波容灾备份。

第一次用Hexo,捣鼓了大半个晚上都生成不了网页,结果最后发现是nodejs14.0.0的问题。

居然因为这种事浪费了四五个小时,也许这就是人生吧。