Caffe：检查失败：错误== cudaSuccess（48 vs. 0）在Jetson TX1上没有内核映像可用于在设备上执行错误

Question

我能够使用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的代码的任何帮助/建议将不胜感激。

Answer 1

确保 MakeFile.config 中关于 jetson 架构的 cuda 拱门编号正确，如下所示：

-gencode arch=compute_72,code=sm_72 \