0

我正在使用 Yolov3 和暗网训练自定义对象检测模型。我最初是从 google colab 开始的,现在我用尽了 colab 的容量。因此,我迁移到 AWS 以访问 GPU。在 Colab 工作期间,我没有遇到任何重大问题。make 命令运行良好,但现在当我尝试在 AWS 服务器上恢复模型的训练时,我在构建暗网时遇到错误,我已将 Makefile 修改为:

GPU = 1
CUDNN = 1
OpenCV = 1

当我设置 GPU=0 时,它工作正常,但是当我设置 GPU=1 时,我收到以下错误。请找到随附的屏幕截图。

看了其他博客后,我什至把 CUDA 版本从 10.0 降级到了 8.0。

请帮我解决这个问题。提前致谢。

请单击下面的链接以查找错误的屏幕截图。 AWS-Makefile-错误

4

1 回答 1

0

问题是 CUDA 版本与暗网不匹配。升级到最新版本——CUDA 11.3后,我的问题解决了!

于 2021-06-17T12:26:05.920 回答