使用mxnet库报错以及解决方案
错误1 Compile with USE_CUDA=1 to enable GPU usage问题原因:一般是因为你在程序中使用了cuda,但是实际上安装了不是cuda版本的mxnet库。解决方法:你需要看下你安装的mxnet库是否是cuda版本的:pip list | grep mxnet如果不是cuda版本的,就需要重新安装下cuda版本的mxnet,需要先卸载原有的,并安装对应cuda版本的
·
错误1 Compile with USE_CUDA=1 to enable GPU usage
问题原因:
一般是因为你在程序中使用了cuda,但是实际上安装了不是cuda版本的mxnet库。
解决方法:
你需要看下你安装的mxnet库是否是cuda版本的:
pip list | grep mxnet
如果不是cuda版本的,就需要重新安装下cuda版本的mxnet,需要先卸载原有的,并安装对应cuda版本的mxnet,例如你的环境中cuda是10.1的,你就安装mxnet-cu101,类似这样:
# 先卸载
pip uninstall mxnet
# 再安装
pip install mxnet-cu101
错误2 OSError: libnccl.so.2: cannot open shared object file: No such file or directory
问题原因:
安装mxnet版本过高
解决方法:
看下你的mxnet版本,然后降低一个版本试下,如我安装了1.8.0版本的mxnet就会出现这个问题,安装1.7.0就没了:
pip install mxnet-cu101==1.7.0
更多推荐

所有评论(0)