我正在使用 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-错误