当前位置: 代码迷 >> 综合 >> ValueError: can‘t optimize a non-leaf Tensor 数据在cpu 和gpu之间转换的时候涉及到requires_grad和is_leaf的趣事
  详细解决方案

ValueError: can‘t optimize a non-leaf Tensor 数据在cpu 和gpu之间转换的时候涉及到requires_grad和is_leaf的趣事

热度:82   发布时间:2024-03-08 02:08:44.0

当我生成一个可导类型的cpu张量的时候,判断其是否为叶子节点,显示为True ,如下面的代码所示:

x = torch.randn(750,2048,requires_grad=True)
print(x.is_leaf)#输出为:True

但是当我直接把这个生成的cpu类型迁移到gpu上时,有意思的事情发生了:

x = torch.randn(750,2048,requires_grad=True).cuda()
print(x.is_leaf)#输出为 False

很蒙蔽吧,那么如何让一个生成的张量迁移到gpu上,并且在gpu上也为叶子节点呢?先将数据迁移到gpu上,然后再设置gpu上面的张量的requires_grad为True ,这样就可以了:

x = torch.randn(750,2048).cuda()
x.requires_grad = True
print(x.is_leaf)#输出结果为: True

 

  相关解决方案