python程序中报错数据在CPU上解决

于 2023-02-06 10:52:59 发布
阅读量102 收藏
点赞数 1
文章标签： python pytorch
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/yu169675/article/details/128897998
版权

之前写程序时遇到的问题，总是报错数据在cpu上：
可以挨个查看每个数据它在那个设备上：

print(datas.device)

然后使用这种方法解决的：

datas=nn.Parameter(datas)
datas=datas.cuda()

OK了！

yu169675
关注
  • 1
    点赞
  • 踩
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值