import torch
device = torch.device('cuda:0')
x = torch.tensor([1,2,3])
x = x.to(device)
import torch
x = torch.tensor([1,2,3])
x = x.cuda(0)

下面的代码是不能改变原数据部署的:

import torch
device = torch.device('cuda:0')
x = torch.tensor([1,2,3])
x.to(device)
import torch
x = torch.tensor([1,2,3])
x.cuda(0)

只能将数据暂时转移到cuda:0上,当这一行代码运行完毕,x又会重新回到CPU上

Logo

北京人形旗下天工造物具身智能开源社区,聚焦具身天工与慧思开物两大平台

更多推荐