torch 一、判断是否支持GPU 一、判断是否支持GPU # torch.cuda.is_available()判断是否支持GPU if torch.cuda.is_available(): # 创建一个张量a a = torch.Tensor([[2, 3], [4, 8]]) a_cuda = a.cuda() print(a_cuda)