pytorchGPU张量运算

CUDA（Compute Unified Device Architecture），是显卡厂商NVIDIA推出的运算平台。 CUDA™是一种由NVIDIA推出的通用并行计算架构，该架构使GPU能够解决复杂的计算问题。 它包含了CUDA指令集架构（ISA）以及GPU内部的并行计算引擎。

import torch

x=torch.rand(5,3)
y=torch.ones(5,3)
if torch.cuda.is_available():#available可获得
x=x.cuda()
y=y.cuda()
print(x+y)

tensor([[1.7949, 1.8783, 1.2017],
        [1.5141, 1.4304, 1.6760],
        [1.3579, 1.9478, 1.0664],
        [1.7044, 1.6285, 1.8083],
        [1.6159, 1.2009, 1.7651]], device='cuda:0')
#device='cuda:0'是可以将运算结果存储到GPU上 
device
n.装置; 仪器; 器具