使用mxnet库报错以及解决方案

导读:本篇文章讲解 使用mxnet库报错以及解决方案,希望对大家有帮助,欢迎收藏,转发!站点地址:www.bmabk.com

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

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

(0)
seven_的头像seven_bm

相关推荐

发表回复

登录后才能评论
极客之音——专业性很强的中文编程技术网站,欢迎收藏到浏览器,订阅我们!