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)