Pytorch细节记录

Pytorch细节记录

获取tensor所在gpu

	device = data.device

将tensor转移到指定gpu上

	a = torch.randn(3,4,device='cuda:0')  # 创建gpu0上的tensor
	a = a.to('cuda:1')  # 转移到gpu1上
	a = a.cuda(1)  # 转移到gpu1
	b = torch.randn(3,4,device=a.device)  # 创建一个和a在一个设备上的tensor

你可能感兴趣的:(pytorch,深度学习,python)