【环境】Win10
一、想在GPU上运行MXNET,报错如下:
MXNetError: [23:23:48] src/storage/storage.cc:xxx: Compile with USE_CUDA=1 to enable GPU usage
【原因】安装的是cpu版的mxnet,不是gpu版的
【解决】卸载原来cpu版本的mxnet(如果猜的没错,你是通过pip install mxnet简单粗暴安装的),安装gpu版的mxnet
安装哪个gpu版本的mxnet呢?看你电脑上装了哪个版本的cuda,(ubuntu在/usr/local/下可以看到)。如果是cuda10.0版,就安装mxnet-cu100(否则还会报错,报错内容见后文)。cuda9.0就是mxnet-cu90。其他的如cuda8.0可能就安装mxnet-cu80吧,没怎么试过。解决过程如下:
# 管理员模式进入cmd
C:\WINDOWS\system32># 卸载cpu版mxnet
C:\WINDOWS\system32> pip uninstall mxnet# 安装gpu版mxnet
C:\WINDOWS\system32> pip install mxnet-cu100
如果版本安的不对,使用mxnet的时候还会报类似如下的错误:
File "f:\python36\lib\ctypes_init_.py", line 348, in init
self._handle = _dlopen(self._name, mode)OSError: [WinError 126] 找不到指定的模块。`