darknet训练出现：CUDA Error: invalid device symbol

最新推荐文章于 2024-05-08 08:37:12 发布

xunan003

最新推荐文章于 2024-05-08 08:37:12 发布

阅读量3.2k

点赞数

CC 4.0 BY-SA版权

分类专栏： Darknet

本文链接：https://blog.youkuaiyun.com/xunan003/article/details/98500413

Darknet 专栏收录该内容

3 篇文章

订阅专栏

可能的原因：显卡计算能力不匹配，未使用CDNN

修改：

makefile：

GPU=1
CUDNN=1
OPENCV=0
OPENMP=0
DEBUG=0

ARCH= -gencode arch=compute_30,code=sm_30 \
      -gencode arch=compute_35,code=sm_35 \
      -gencode arch=compute_50,code=[sm_50,compute_50] \
      -gencode arch=compute_61,code=[sm_61,compute_61]
      #-gencode arch=compute_20,code=[sm_20,sm_21] #\ This one is deprecated?

ARCH中的数字对应显卡计算能力，获取显卡计算能力可参考链接：https://developer.nvidia.com/cuda-gpus