用gpu跑gluon,以及解决It was only initialized on [cpu(0)].

  1, 数据要设置在gpu上存储,如:

a = nd.array([1,2,3], ctx=mx.gpu())

  2,不只是第一次权重初始化的时候需要设置ctx,在后来每次调用intialize()函数的时候也要设置ctx,具体设置如下:

net.collect_params().initialize(force_reinit=True, ctx=mx.gpu())

  3,iterator里出来的数据也要设置ctx。如:

for data, label in data_iter_train:
    data = data.as_in_context(ctx)
    label = label.as_in_context(ctx)

  最开始的时候,我出现报错:
  

RuntimeError: Parameter sequential0_dense0_weight was not initialized on context cpu(0). It was only initialized on [gpu(0)]

  然后按照上面那3点设置好之后,就不报错了。

  • 1
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值