terminate called after throwing an instance of dmlc::Error

在python中使用MxNet时在代码

 self.module.bind(data_shapes=[data_name_shape],
                         for_training=False)

报出一下错误

terminate called after throwing an instance of 'dmlc::Error'
  what():  [07:08:37] src/storage/storage.cc:119: Compile with USE_CUDA=1 to enable GPU usage
Stack trace:

主要原因是使用

pip install mxnet

默认安装得不是gpu版得,应该安装对应cuda版本得mxnet就应该没有问题了。

cuda10.1得下载地址

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值