pyTorch学习基础一选择CPU和GPU

CPU： 数据读取、数据预处理

GPU：参数计算、推理 

import torch

# 张量在CPU上计算
dev_cpu=torch.device("cpu") #指定CPU计算
a= torch.tensor([2,2],dtype=torch.float32,device=dev_cpu) #张量选择设备
print(a)

# 张量在GPU上计算
dev_gpu=torch.device("cuda") #指定GPU计算
b= torch.tensor([2,2],dtype=torch.float32,device=dev_gpu) #张量选择设备
print(b)