yolo导出模型默认cpu,
在导入模型后再cpu下可以运行,gpu下出错:Microsoft C++异常:c10::Error
需要改动exporter.py中的select_device('cpu' if self.args.device is None else self.args.device)
把cpu改为0,即为用gpu0运行生成,再生成时显示
显示为cuda0后,导出的为gpu下,
yolo导出模型默认cpu,
在导入模型后再cpu下可以运行,gpu下出错:Microsoft C++异常:c10::Error
需要改动exporter.py中的select_device('cpu' if self.args.device is None else self.args.device)
把cpu改为0,即为用gpu0运行生成,再生成时显示
显示为cuda0后,导出的为gpu下,