我能够使用CUDA 9.0和Open CV 3在Nvidia Jetson TX1板上成功编译Caffe。
但是，当我运行以下命令来测试Caffe时：
build/tools/caffe time --model=models/bvlc_alexnet/deploy.prototxt --gpu=0
我收到以下错误：
F0712 23:05:53.664676 28580 im2col.cu:61] Check failed: error == cudaSuccess (48 vs. 0) no kernel image is available for execution on the device
如果我删除了--gpu=0标志，我再也看不到错误了。
对于我如何获得使用GPU的代码的任何帮助/建议，将不胜感激。
答案 0 :(得分：0)
确保关于Jetson架构，在MakeFile.config中具有正确的cuda拱门编号，如下所示：
-gencode arch=compute_72,code=sm_72 \