[人工智能]如何使用to(device)以及如何解决AttributeError: ‘torch.device‘ object has no attribute ‘_apply‘

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

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)

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