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

本文针对CUDA环境下编译问题与mxnet库版本不匹配的常见错误,提供了解决方法。首先,检查是否安装了正确的cuda版本mxnet;其次,遇到libnccl.so.2找不到的错误,可能是由于mxnet版本过高,建议降低到1.7.0。
摘要由CSDN通过智能技术生成

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

参考

  • 4
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值