前言
安装好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~~
参考
2. 解决方法;
完