yolov5跑voc数据集看看效果,下载的 ## conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch ##,但是看gpu的占用率很低很低,相反cpu占用率很高,这个到底是不是在用gpu跑啊,有人知道吗?
运行中:
大概率是cpu再跑,你可以打开anaconda,激活环境,输入python,然后输入指令>>> import torch
torch.cuda.is_available(),看看有没有报错,如果有有可能是安装的cuda的问题,如果没有报错,看看代码是不是参数默认使用cpu
你debug一下 看看相关变量在不在gpu上就好了. 不在的话就移到gpu上 具体可百度 一句代码的事
建议你看下这篇博客yolov5(GPU)配置