google-colaboratory - CUDA 错误：未检测到支持 CUDA 的设备：错误的文件描述符 Google Colab

Question

我将通过以下代码训练我的 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 进行了尝试。

有没有办法解决这个问题？

标签： google-colaboratoryobject-detectionyolodarknet

Answer 1

TPU 未启用 CUDA

尝试以下代码片段以确保

device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
print(device)