python 查看torch第哦啊用gpu还是cpu

在Python中，如果你使用PyTorch库进行深度学习开发，并想检查张量是否在GPU上运行，你可以使用`.to()`函数将张量移动到设备，然后检查其设备类型。如果你想查看已经存在的张量是在CPU还是GPU上，可以使用`.is_cuda`属性或者`.device`属性。

1. `.is_cuda` 属性会直接告诉你张量是否位于GPU上：

   tensor = torch.tensor(...).cuda()  # 或者 tensor = torch.randn(10, 10)
   if tensor.is_cuda:
       print("Tensor is on GPU")
   else:
       print("Tensor is on CPU")

2. `.device` 属性会返回张量所在的设备，如果在GPU上，字符串形式将是 "cuda"：

   device = tensor.device
   if device.type == 'cuda':
       print("Tensor is on GPU")
   elif device.type == 'cpu':
       print("Tensor is on CPU")