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

时间:2018-07-13 21:56:06

标签: caffe

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

1 个答案:

答案 0 :(得分:0)

确保关于Jetson架构,在MakeFile.config中具有正确的cuda拱门编号,如下所示:

-gencode arch=compute_72,code=sm_72 \