pytorch测试gpu

import torch

t = torch.tensor([2, 3])
print(t)
x=torch.randn(3,4)
print(x)
# x_gpu=x.cuda(0)
x_gpu2=x.cuda()
print(x_gpu2)
x_cpu=x_gpu2.cpu()
print(x_cpu)
x_arr=x_cpu.cpu().numpy()
print(x_gpu2)
print(x_arr)