YOLO 重启后不使用 GPU

问题描述

在重新启动之前,我可以很好地将 YOLO 与 GPU 配合使用,如下所示。

PS C:\Users\darknet> ./darknet.exe detector train data/obj.data yolo-obj.cfg backup/yolo-obj_final.weights -clear
 CUDA-version: 10010 (11000),cuDNN: 7.6.5,CUDNN_HALF=1,GPU count: 1
 CUDNN_HALF=1
 OpenCV version: 4.3.0
yolo-obj
 0 : compute_capability = 750,cudnn_half = 1,GPU: GeForce GTX 1660
net.optimized_memory = 0
mini_batch = 8,batch = 64,time_steps = 1,train = 1

然而,我的电脑因为不知道的原因死机了,我重新启动了电脑,它突然无法检测到 GPU。即使包含 -gpus 0 命令。

PS C:\Users\darknet> .\darknet.exe detector train data/obj.data yolo-obj.cfg backup/yolo-obj_final.weights -gpus 0
 GPU isn't used
 Used AVX
 Used FMA & AVX2
 OpenCV version: 4.4.0
0
yolo-obj
mini_batch = 8,train = 1

我尝试重新安装 CUDA,但仍然无法正常工作。 我该如何解决这个问题?

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)