错误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