【err】开启Persistence-M模式-Check failed: err == CUBLAS_STATUS_SUCCESS (1 vs. 0) : Create cublas handle failed

前言

安装好CUDA、CUDNN、NVIDIA driver之后,使用mxnet框架的时候出现该错误,本文记录该问题的解决方法。

环境

ubuntu 16.04

MxNet

Cuda9.0

Nvidia driver 384

error

terminate called after throwing an instance of 'dmlc::Error'
what(): [::] /home/travis/build/dmlc/mxnet-distro/mxnet-build/3rdparty/mshadow/mshadow/./stream_gpu-inl.h:: Check failed: err == CUBLAS_STATUS_SUCCESS ( vs. ) : Create cublas handle failed
Stack trace:
*************
[bt] () ~/miniconda3/bin/../lib/libstdc++.so.(+0xb8678) [0x7f8622101678]
[bt] () /lib/x86_64-linux-gnu/libpthread.so.(+0x76ba) [0x7f86731206ba]
[bt] () /lib/x86_64-linux-gnu/libc.so.(clone+0x6d) [0x7f8672e5641d]
Aborted (core dumped)

解决方法

开始的时候以为是driver没有安装好,但是使用nvidia-smi之后可以显示GPU信息,只是疑惑Persistence-M为什么是off呢。。。然后就将其状态改为ON试试,就可以了。

nvidia-smi -pm 

或者

sudo nvidia-persistenced --persistence-mode

博主使用的是第一个命令行,可以试试第二种。

其实,最重要的是要明白nvidia-smi的命令。

哈哈哈,终于解决这个遗留问题啦。。。开心O(∩_∩)O~~

参考

1.GPU状态监控nvidia-smi命令详解

2. 解决方法

上一篇:Redis——Linux(centos7.x)下Redi和PHP Redis插件安装——【一】


下一篇:Check failed: status == CUBLAS_STATUS_SUCCESS (11 vs. 0) CUBLAS_STATUS_MAPPING_ERROR