pytorch使用GPU进行加速
2020-04-30 本文已影响0人
一位学有余力的同学
device=torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
model=model.to(device)
x=x.to(device)
y=y.to(device)
数据都是GPU形式的，如果想要使用numpy对GPU运算的结果进行处理，需要先把GPU的数据转换到CPU上的tensor形式，然后在转换成numpy形式。
data.cpu().numpy()
