search

CVUsers / Smoke-Detect-by-YoloV5

Yolov5 real time smoke detection system
GNU General Public License v3.0
434 stars 119 forks source link

设置了device， 但是没有用gpu #10

Open 15926273249 opened 3 years ago

15926273249 commented 3 years ago

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

CVUsers commented 3 years ago

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