项目场景:
环境:ubuntu16.04
cuda版本10.0
运行caffe时
例如:sudo make runtest
问题描述:
Check failed: error == cudaSuccess (48 vs. 0) no kernel image is available for execution on the device
原因分析:GPU算力不对应
改正Makefile.config 里面的CUDA_ARCH
第一步:在caffe目录下
sudo gedit Makefile.config
修改Makefile.config 里面的CUDA_ARCH
CUDA_ARCH := -gencode arch=compute_30,code=sm_30 \
-gencode arch=compute_35,code=sm_35 \
-gencode arch=compute_50,code=sm_50 \
-gencode arch=compute_52,code=sm_52 \
-gencode arch=compute_60,code=sm_60 \
-gencode arch=compute_61,code=sm_61 \
-gencode arch=compute_61,code=compute_61
如图所示:
第二步:必须做,不做编译绝对不通过
sudo make clean
第三步:
sudo make all
第四步:
sudo make runtest
编译成功