對於「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 有這樣一句代碼,我們需要關注-
相關文章
相關標籤/搜索