【发布时间】:2021-09-08 06:30:41
【问题描述】:
我将通过以下代码训练我的 yolo 模型。
! darknet/darknet detector train custom_data/labelled_data.data darknet/cfg/yolov3_custom.cfg custom_weight/darknet53.conv.74 -dont_show
然后显示如下错误
CUDA status Error: file: ./src/dark_cuda.c : () : line: 39 : build time: Sep 8 2021 - 06:14:34
CUDA Error: no CUDA-capable device is detected
Darknet error location: ./src/dark_cuda.c, check_error, line #70
CUDA Error: no CUDA-capable device is detected: Bad file descriptor
首先我尝试了使用 GPU 的代码。但是超过 GPU 的限制后,我用 TPU 进行了尝试。
有没有办法解决这个问题?
【问题讨论】:
-
您不能使用 TPU 运行 CUDA 代码。它需要一个 CUDA GPU。你想做的事是不可能的
标签: google-colaboratory object-detection yolo darknet