【问题标题】:CUDA Error: no CUDA-capable device is detected: Bad file descriptor Google ColabCUDA 错误:未检测到支持 CUDA 的设备:错误的文件描述符 Google Colab
【发布时间】: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


【解决方案1】:

TPU 未启用 CUDA

尝试以下 sn-p 以确保

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

【讨论】:

    猜你喜欢
    • 1970-01-01
    • 1970-01-01
    • 2012-10-14
    • 2012-10-31
    • 2023-01-04
    • 1970-01-01
    • 2020-12-30
    • 1970-01-01
    • 1970-01-01
    相关资源
    最近更新 更多