首页
学习
活动
专区
工具
TVP
发布
精选内容/技术社群/优惠产品,尽在小程序
立即前往
您找到你想要的搜索结果了吗?
是的
没有找到

PyTorch代码调试利器: 自动print每行代码的Tensor信息

比如你可能在代码的第三行用 torch.zeros 新建了一个 CPU tensor, 然后这个 tensor 进行了若干运算,全是在 CPU 上进行的,一直没有报错,直到第十行需要跟你作为输入传进来的...例子1 比如说我们写了一个非常简单的函数: def myfunc(mask, x): y = torch.zeros(6) y.masked_scatter_(mask, x)...cuda:0> 21:41:42.941668 call 5 def myfunc(mask, x): 21:41:42.941834 line 6 y = torch.zeros...这一行对应代码中的 y = torch.zeros(6)。于是我们意识到,在使用 torch.zeros 的时候,如果不人为指定设备的话,默认创建的 tensor 是在 CPU 上的。...我们把这一行改成 y = torch.zeros(6, device= cuda ),这一行的问题就修复了。

1K10

PyTorch代码调试利器: 自动print每行代码的Tensor信息

比如你可能在代码的第三行用 torch.zeros 新建了一个 CPU tensor, 然后这个 tensor 进行了若干运算,全是在 CPU 上进行的,一直没有报错,直到第十行需要跟你作为输入传进来的...例子1 比如说我们写了一个非常简单的函数: def myfunc(mask, x): y = torch.zeros(6) y.masked_scatter_(mask, x)...cuda:0> 21:41:42.941668 call 5 def myfunc(mask, x): 21:41:42.941834 line 6 y = torch.zeros...这一行对应代码中的 y = torch.zeros(6)。于是我们意识到,在使用 torch.zeros 的时候,如果不人为指定设备的话,默认创建的 tensor 是在 CPU 上的。...我们把这一行改成 y = torch.zeros(6, device='cuda'),这一行的问题就修复了。

1.1K20
领券