火车Caffe模型:检查失败:错误== cudaSuccess(8对0)无效的设备功能

时间:2018-06-06 14:16:07

标签: caffe nvidia nvidia-digits nvidia-docker nvidia-jetson

我正在浏览此tutorial的检测部分,以使用DIGITS在我的Jetson TX2上部署模型。我在DIGITS上创建了一个数据集,然后我试图训练一个模型,但我不断得到"错误== cudaSuccess(8对0)无效的设备功能"错误。我有一个NVIDIA 950M(计算能力5.0),我确保在安装NVcaffe时我的Makefile.config包含它:

# CUDA architecture setting: going with all of them.
# For CUDA < 6.0, comment the *_50 lines for compatibility.
CUDA_ARCH := -gencode arch=compute_20,code=sm_20 \
    -gencode arch=compute_20,code=sm_21 \
    -gencode arch=compute_30,code=sm_30 \
    -gencode arch=compute_35,code=sm_35 \
    -gencode arch=compute_50,code=sm_50 \
    -gencode arch=compute_50,code=compute_50

我有CUDA 9.0,cuDNN 7.0.5,正如我所说,我的GPU是带有sm_50的NVIDIA GTX 950M,当我运行cmake来构建NVcaffe时也会检测到它。我真的不能理解什么是错的,所以任何帮助都会受到高度赞赏。

Here是我尝试在DIGITS上训练模型时生成的caffe_output.log文件的Pastebin链接(错误报告在文件底部)。

提前致谢

编辑1 我也注意到DIGITS检测到了电路板。我截取了界面的屏幕截图,显示检测到GPU的时间为半秒:

Just before crashing

编辑1 我刚刚使用CPU_ONLY模式再次安装了caffe,但我遇到了完全相同的问题,DIGITS正在尝试使用我的GPU,所以我认为这个问题与之前的caffe安装有关。我不得不说,因为没有&#39; make uninstall&#39;我可以运行的命令我刚刚删除了caffe文件夹,我重复了安装过程git再次克隆了repo。

0 个答案:

没有答案