Avar=var.to("cuda:0")
Bvar=var.to("cuda:1")
Cvar=var.cuda()
Dvar=var.cuda()
在pytorch中,如果主机有1块TitanX显卡,以下哪个选项中的代码都可以将变量名为var的tensor放在GPU上运行
在pytorch中,如果主机有1块TitanX显卡,以下哪个选项中的代码都可以将变量名为var的tensor放在GPU上运行()
首页
每日一练
打赏一下
浏览记录