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

本文介绍了如何在GPU上存储数据、初始化权重以及确保所有操作都在GPU环境中完成的方法。文章详细解释了在深度学习中避免运行时错误的三个关键步骤,并通过实例说明了如何正确地将数据和模型迁移到GPU。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

  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点设置好之后,就不报错了。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值