对于“cudaCheckError() failed : invalid device function”错误的解决方法

我的问题出现在训练faster_rcnn的时候,后来经过在网上寻找解决方案,找到的原因是gpu的计算能力与配置的不符 解决方法: 在make.sh里面 nvcc -c -o roi_pooling.cu.o roi_pooling_kernel.cu \ -D GOOGLE_CUDA=1 -x cu -Xcompiler -fPIC -arch=sm_52 有这样一句代码,我们需要关注-
相关文章
相关标签/搜索