YOLO-v5 icon indicating copy to clipboard operation
YOLO-v5 copied to clipboard

Published 20 hours ago • verified publisher icon DataXujing

test

Open geng007 opened this issue 5 years ago • 4 comments
trafficstars

您好！ image test使用GPU，bug，是代码不支持吗～

geng007 avatar Jul 02 '20 03:07 geng007

请检查你的pytorch可以正常调用GPU，怀疑你的pytorch编译的CUDA版本和你机器装的CUDA版本不一致，导致pytorch不能使用GPU

DataXujing avatar Jul 02 '20 04:07 DataXujing 

import torch
print(torch.cuda.is_available())

应该是返回:True 的请自查

DataXujing avatar Jul 02 '20 04:07 DataXujing

请指定使用的device

DataXujing avatar Aug 19 '20 08:08 DataXujing

请指定使用的device

问题解决了，model和x都是cuda，报错是因为使用了htop库，我也不知道为什么会有这个库，用pip3 uninstall删掉就好了。 参考 https://github.com/ultralytics/yolov5/issues/210 感谢作者回复~

TomSirLiu avatar Aug 19 '20 08:08 TomSirLiu