哥避专题

【Xiang哥避坑指南】YOLOV5只在CPU跑不在GPU跑的问题。

电脑明明有GPU,深度学习视觉代码也运行起来了,为什么只在CPU跑,GPU并不占用呢,可能有的同学会有这种经历。 本人就是被困扰了一个月,不知道为什么只在CPU跑,最后偶然发现: 那可能就是CUDA、pytorch、torchvison三者版本不对应的问题,也可能是版本下错的问题。 CUDA、pytorch、torchvision版本一定要对应!!! CUDA、pytorch、torchvis