Smoke-Detect-by-YoloV5 icon indicating copy to clipboard operation
Smoke-Detect-by-YoloV5 copied to clipboard

Published 20 hours ago • verified publisher icon CVUsers

设置了device， 但是没有用gpu

Open 15926273249 opened this issue 4 years ago • 1 comments
trafficstars

您好，请问为什么设置了device, 但是运行 detect.py脚本的时候还是显示using cpu

15926273249 avatar Dec 22 '20 09:12 15926273249

device那边设置为cpu才会使用cpu，写0是gpu，如果写0，后还是使用cpu，并且nvidia－smi查看占用显存很少，那么请检查pytorch是否支持cuda torch.cuda.is_available------------------ 原始邮件 ------------------ 发件人: "15926273249"<[email protected]> 发送时间: 2020年12月22日(星期二) 下午5:46 收件人: "CVUsers/Smoke-Detect-by-YoloV5"<[email protected]>; 抄送: "Subscribed"<[email protected]>; 主题: [CVUsers/Smoke-Detect-by-YoloV5] 设置了device， 但是没有用gpu (#10)

CVUsers avatar Dec 22 '20 10:12 CVUsers