如何检查模型是否在 CUDA 上？

Question

我想检查模型是否在 CUDA 上。怎么做？

import torch
import torchvision
model = torchvision.models.resnet18()
model.to('cuda')

model.is_cuda()不工作的接缝。

Answer 1

这段代码应该这样做：

import torch
import torchvision
model = torchvision.models.resnet18()
model.to('cuda')
next(model.parameters()).is_cuda

出去：

True

注意里面没有is_cuda()方法nn.Module。还要注意model.to('cuda')是相同的，model.cuda()并且两者都是到位的。

另一方面，移动data.to('cuda')不是到位的，您通常会调用：

True

将数据移动到 CUDA。