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

43 篇文章 2 订阅
在Win10环境下,运行MXNET时遇到GPU不支持的错误。原因是安装了CPU版MXNET,而非GPU版。解决方法是卸载现有CPU版本,根据CUDA版本(如cuda10.0对应mxnet-cu100)正确安装GPU版MXNET。安装错误的CUDA版本会导致运行时错误。
摘要由CSDN通过智能技术生成

【环境】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\s
评论 12
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值