1 查看tensor在那个gpu id/device上
print(data.device) 或者在调试模式查看模型或tensor的device属性
2 指定模型device
model.to(device)
注:如果用.cuda() 会将模型load到默认gpu:0上
3 load checkpoints到指定device
torch.load(checkpoint_path, map_location=args.device)
其它情况及多gpu情况待更新
1 查看tensor在那个gpu id/device上
print(data.device) 或者在调试模式查看模型或tensor的device属性
2 指定模型device
model.to(device)
注:如果用.cuda() 会将模型load到默认gpu:0上
3 load checkpoints到指定device
torch.load(checkpoint_path, map_location=args.device)
其它情况及多gpu情况待更新