首页
Python
Java
前端
数据库
Linux
Chatgpt专题
开发者工具箱
哥避专题
【Xiang哥避坑指南】YOLOV5只在CPU跑不在GPU跑的问题。
电脑明明有GPU,深度学习视觉代码也运行起来了,为什么只在CPU跑,GPU并不占用呢,可能有的同学会有这种经历。 本人就是被困扰了一个月,不知道为什么只在CPU跑,最后偶然发现: 那可能就是CUDA、pytorch、torchvison三者版本不对应的问题,也可能是版本下错的问题。 CUDA、pytorch、torchvision版本一定要对应!!! CUDA、pytorch、torchvis
阅读更多...