pytorch使用中torch.uint8转换torch.float()卡住

pytorch使用中torch.uint8转换torch.float()卡住

问题描述:python3.6.8使用yolov5目标检测项目时,发现torch.uint8转换torch.float()卡住,即在代码
im = im.half() if self.model.fp16 else im.float()
这行卡死,也不报错,可能是torch在使用内存过程中起冲突了。

解决方案:将python升级到最新的3.10就完美解决。

注意:1、其它版本的python没试过,可能升级到3.7就可以了,但没试过。

你可能感兴趣的:(深度学习,深度学习,python,目标检测,pytorch)