当前位置: 代码迷 >> 综合 >> src/storage/storage.cc: Compile with USE_CUDA=1 to enable GPU usage【MXNET GPU版】
  详细解决方案

src/storage/storage.cc: Compile with USE_CUDA=1 to enable GPU usage【MXNET GPU版】

热度:43   发布时间:2023-12-15 17:33:07.0

【环境】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] 找不到指定的模块。`

 

  相关解决方案