如何使用to(device)以及如何解决AttributeError: ‘torch.device‘ object has no attribute ‘_apply‘_x, y = x.to(device), y.to(device) attributeerror:

作者：IT小白 | 2024-06-14 03:07:54

踩

x, y = x.to(device), y.to(device) attributeerror: 'int' object has no attrib

如果是类的话，就一定要实例化才能用：

class X():

def __init__(self):

super(X, self).__init__()

DEVICE = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")

X = X().to(DEVICE)

如果没有实例化的话就会报错：AttributeError: 'torch.device' object has no attribute '_apply'

如果是数据的话就可以直接用：

img = img.to(DEVICE)

（此博客仅为了记录自己踩坑过程）

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/IT小白/article/detail/716019