错误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
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 举报,一经查实,本站将立刻删除。
文章由极客之音整理,本文链接:https://www.bmabk.com/index.php/post/121191.html