Detectron2虚拟机预测推断时无输出,卡退问题如何解决?

摘要:环境 虚拟机 Ubuntu 18.04 问题描述 在实体机上运行没有任何问题,在虚拟机上运行到预测图片这一步直接卡退,没有任何输出。 ... predictor = DefaultPredictor(cfg) outputs = predi
环境 虚拟机 Ubuntu 18.04 问题描述 在实体机上运行没有任何问题,在虚拟机上运行到预测图片这一步直接卡退,没有任何输出。 ... predictor = DefaultPredictor(cfg) outputs = predictor(imgs) # 这一步直接卡退 解决过程 使用 pdb 库进入 detectron2 库源码里断点调 bug,最终发现 bug 出现在 torch.Tensor.max() 方法这一步,报错内容: Illegal instruction (core dumped) 搜索发现是因为 KVM 虚拟机不支持 AVX(Advanced Vector Extensions)。联系虚拟机提供方开启了此功能,问题解决。