导入mxnet时指令非法(内核转储)

时间:2019-01-24 13:58:33

标签: python-2.7 mxnet

调用import mxnet时出现“非法指令(内核已转储)”异常。我正在使用CUDA 9.0,并使用anaconda安装了mxnet。我的python版本是2.7。附带说明,我的计算机上还安装了CUDA 10.0。

任何帮助将不胜感激。

1 个答案:

答案 0 :(得分:1)

“非法指令(转储内核)”通常是由于MXNet加载了与最初使用CUDA,OpenMP等构建的共享库不同的共享库。一种解决方案是直接从源代码构建在您自己的系统上。

例如:

    git clone --recursive https://github.com/apache/incubator-mxnet.git
    cd incubator-mxnet
    echo "USE_OPENCV = 1" >> ./config.mk
    echo "USE_BLAS = openblas" >> ./config.mk
    echo "USE_CUDA = 1" >> ./config.mk
    echo "USE_CUDA_PATH = /usr/local/cuda" >> ./config.mk
    echo "USE_CUDNN = 1" >> ./config.mk
    make -j $(nproc)

完整的安装说明可在MXNet网站上找到:http://mxnet.incubator.apache.org/versions/master/install/index.html?platform=Linux&language=Python&processor=CPU