今天小编就为大家分享一篇pytorch查看torch.Tensor和model是否在CUDA上的实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

今天训练faster R-CNN时，发现之前跑的很好的程序（是指在运行程序过程中，显卡利用率能够一直维持在70%以上），今天看的时候，显卡利用率很低，所以在想是不是我的训练数据torch.Tensor或者模型model没有加载到GPU上训练，于是查找如何查看tensor和model所在设备的命令。 import torch import torchvision.models as models model=models.vgg11(pretrained=False) print(next(mo