Pytorch 如何检查CUDA上是否有模型？

我想检查一下CUDA上是否有型号。怎么做

导入火炬
进口火炬视觉
model=torchvision.models.resnet18（）
型号。至（‘cuda’）

---

model.is_cuda（）

的接缝不起作用。

此代码应该可以：

导入火炬
进口火炬视觉
model=torchvision.models.resnet18（）
型号。至（‘cuda’）
下一个（model.parameters（））。是_cuda

输出：

注意

nn.Module

中没有

is_cuda（）

方法。 另请注意

model.to（'cuda'）

与

model.cuda（）

相同，并且两者都在同一位置

另一方面，将

数据移动到（'cuda'）

是不到位的，您通常会调用：

data=data.to（'cuda'））

---

要将数据移动到CUDA。

此代码应执行以下操作：

导入火炬
进口火炬视觉
model=torchvision.models.resnet18（）
型号。至（‘cuda’）
下一个（model.parameters（））。是_cuda

输出：

注意

nn.Module

中没有

is_cuda（）

方法。 另请注意

model.to（'cuda'）

与

model.cuda（）

相同，并且两者都在同一位置

另一方面，将

数据移动到（'cuda'）

是不到位的，您通常会调用：

data=data.to（'cuda'））

将数据移动到CUDA

True