Getting OSError: libnccl.so.2 while importing mxnet

Getting OSError: libnccl.so.2 while importing mxnet

我的配置是:

gpu: Nvidia GeForce MX450 with 2GB Graphics Card
Cuda version: 10.1
MxNet Version: mxnet-cu101: 1.8.0

尝试使用 import mxnet mxnet 时出现以下错误:

OSError: libnccl.so.2: cannot open shared object file: No such file or directory

过去 2 年我一直在使用 mxnet,但以前从未遇到过这个问题。据我所知,mxnet 将它们作为依赖项自行安装。然而这次没有发生。

这只发生在mxnet-cu101==1.8.0版本。我降级到 mxnet-cu101==1.7,问题得到解决。

或者您可以升级到更新的版本:

pip install mxnet-cu110==2.0.0a0

这解决了问题