【发布时间】:2021-04-01 17:05:48
【问题描述】:
我想用 gpu 安装 mxnet。并使用pip install mxnet-cu90。但是,当我导入包时,我遇到了这个错误:
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "C:\Users\Anuj\anaconda3\envs\py36\lib\site-packages\mxnet\__init__.py", line 24, in <module>
from .context import Context, current_context, cpu, gpu, cpu_pinned
File "C:\Users\Anuj\anaconda3\envs\py36\lib\site-packages\mxnet\context.py", line 24, in <module>
from .base import classproperty, with_metaclass, _MXClassPropertyMetaClass
File "C:\Users\Anuj\anaconda3\envs\py36\lib\site-packages\mxnet\base.py", line 213, in <module>
_LIB = _load_lib()
File "C:\Users\Anuj\anaconda3\envs\py36\lib\site-packages\mxnet\base.py", line 204, in _load_lib
lib = ctypes.CDLL(lib_path[0], ctypes.RTLD_LOCAL)
File "C:\Users\Anuj\anaconda3\envs\py36\lib\ctypes\__init__.py", line 348, in __init__
self._handle = _dlopen(self._name, mode)
OSError: [WinError 126] The specified module could not be found
我尝试通过从this 下载 .whl 文件重新安装,但没有成功。它仍然给出同样的错误。
我正在使用 python 3.6.12。
【问题讨论】:
-
它可能会尝试加载一些您可能需要单独安装的 C/C++ 库(但不能使用
pip) -
好的。我怎样才能知道我必须安装哪个库?
-
错误为您提供
mxnet\base.py的路径,您可以打开它并在代码中查看它尝试加载的库。最后你应该在官方页面上查看mxnet的文档,也许他们描述了你还需要安装什么。 -
我查看mxnet 的官方页面和Get Started 页面(在选择我要安装的版本后)我显示:应该首先安装CUDA。从版本 1.8.0 开始,还应安装 CUDNN 和 NCCL。。它还显示使用
nvcc --version检查已安装的 CUDA -
是的,你完全正确。我确实浏览了他们的文档并下载了 Nvidia cuda 11.2,然后可以运行命令
nvcc --version。我还安装了 cudatoolkit,但它仍然无法正常工作。我还没有听说过 NCCL,我会再调查一下。